github actions tweaks

Author: isabelizimm

.github/workflows/tests.yml

@@ -32,7 +32,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install -e .[dev,torch]
+          python -m pip install -e .[dev,torch,statsmodels]
       - name: Run Tests
         run: |
           pytest -m 'not rsc_test' --cov --cov-report xml

vetiver/handlers/statsmodels.py

@@ -58,7 +58,8 @@ def handler_predict(self, input_data, check_ptype):
         prediction
             Prediction from model
         """
-
+        if not check_ptype:
+            input_data = pd.DataFrame(input_data)
         if isinstance(input_data, pd.DataFrame):
             prediction = self.model.predict(input_data)
         else:

vetiver/server.py

@@ -82,8 +82,7 @@ async def prediction(
 
             @app.post("/predict")
             async def prediction(input_data: Request):
-
-                y = await input_data.json()
+                y = await input_data.body()
                 prediction = self.model.handler_predict(y, check_ptype=self.check_ptype)
 
                 return {"prediction": prediction.tolist()}

vetiver/tests/test_statsmodels.py

@@ -26,9 +26,24 @@ def test_vetiver_build(build_sm):
     api = vetiver.VetiverAPI(build_sm)
     client = TestClient(api.app)
     data = {"B": 0, "C": 0, "D": 0}
-    response = client.post("/predict", json=data)
-    assert response.status_code == 200, response.text
-    assert response.json() == {"prediction": [0.0]}, response.json()
+
+    response = vetiver.predict(endpoint=client, data=data)
+
+    assert response.iloc[0, 0] == 0.0
+    assert len(response) == 1
+
+
+def test_no_ptype(build_sm):
+    v = build_sm
+    v.ptype = None
+    api = vetiver.VetiverAPI(v)
+    client = TestClient(api.app)
+    data = 0, 0, 0
+
+    response = vetiver.predict(endpoint=client, data=data)
+
+    assert response.iloc[0, 0] == 0.0
+    assert len(response) == 1
 
 
 def test_serialize(build_sm):