add github action for testing

.github/workflows/test.yaml

@@ -0,0 +1,42 @@
+name: test
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+
+jobs:
+  test:
+    if: github.event.pull_request.merged == false
+    runs-on: ubuntu-latest
+
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.8", "3.9", "3.10", "3.11"]
+
+    env:
+      PYTHON: ${{ matrix.python-version }}
+
+    steps:
+      - uses: actions/checkout@v3
+
+      - name: set up python ${{ matrix.python-version }}
+        uses: actions/setup-python@v4
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install package
+        run: |
+          pip install ".[dev]"
+
+      - name: Run mypy
+        run: mypy .
+
+      - name: Run black
+        run: |
+          black nrel tests --check
+
+      - name: Python unit tests
+        run: |
+          pytest tests/ -v

nrel/routee/powertrain/estimators/estimator_interface.py

@@ -7,7 +7,6 @@
 from nrel.routee.powertrain.core.features import DataColumn, FeatureSet, TargetSet
 
 
-
 class Estimator(ABC):
     @classmethod
     @abstractmethod

nrel/routee/powertrain/estimators/sklearn/__init__.py

@@ -1 +1 @@
-from .estimator import SKLearnEstimator 
+from .estimator import SKLearnEstimator

nrel/routee/powertrain/trainers/sklearn_random_forest.py

@@ -9,6 +9,7 @@
 from nrel.routee.powertrain.estimators.onnx import ONNX_INPUT_NAME, ONNXEstimator
 from nrel.routee.powertrain.trainers.trainer import Trainer
 
+
 class RandomForestTrainerOutput(Enum):
     ONNX = 1
 
@@ -54,11 +55,9 @@ def inner_train(
         if self.output_type == RandomForestTrainerOutput.ONNX:
             # convert to ONNX
             n_features = len(features.columns)
-            initial_type = [
-                (ONNX_INPUT_NAME, FloatTensorType([None, n_features]))
-            ]
+            initial_type = [(ONNX_INPUT_NAME, FloatTensorType([None, n_features]))]
             onnx_model = to_onnx(
-                rf, 
+                rf,
                 initial_types=initial_type,
             )
 

pyproject.toml

@@ -34,6 +34,11 @@ dev = [
     "boxsdk",
     "jupyter-book",
     "sphinx-book-theme",
+    "types-PyYAML",
+    "types-protobuf",
+    "types-python-dateutil",
+    "types-redis",
+    "types-requests"
 ]
 
 [project.urls]