remove tolist from API

Author: isabelizimm

vetiver/handlers/sklearn.py

@@ -42,4 +42,4 @@ def handler_predict(self, input_data, check_prototype):
         else:
             prediction = self.model.predict([input_data])
 
-        return prediction
+        return prediction.tolist()

vetiver/handlers/statsmodels.py

@@ -51,4 +51,4 @@ def handler_predict(self, input_data, check_prototype):
         else:
             prediction = self.model.predict([input_data])
 
-        return prediction
+        return prediction.tolist()

vetiver/handlers/torch.py

@@ -53,4 +53,4 @@ def handler_predict(self, input_data, check_prototype):
             input_data = torch.tensor(input_data)
             prediction = self.model(input_data)
 
-        return prediction
+        return prediction.tolist()

vetiver/handlers/xgboost.py

@@ -56,4 +56,4 @@ def handler_predict(self, input_data, check_prototype):
 
         prediction = self.model.predict(input_data)
 
-        return prediction
+        return prediction.tolist()

vetiver/server.py

@@ -175,17 +175,17 @@ def vetiver_post(self, endpoint_fx: Callable, endpoint_name: str = None, **kw):
 
             @self.app.post(urljoin("/", endpoint_name), name=endpoint_name)
             async def custom_endpoint(input_data: List[self.model.prototype]):
-                new = endpoint_fx(input_data, **kw)
-                return {endpoint_name: new.tolist()}
+                predictions = endpoint_fx(input_data, **kw)
+                return {endpoint_name: predictions}
 
         else:
 
             @self.app.post(urljoin("/", endpoint_name))
             async def custom_endpoint(input_data: Request):
                 served_data = await input_data.json()
-                new = endpoint_fx(served_data, **kw)
+                predictions = endpoint_fx(served_data, **kw)
 
-                return {endpoint_name: new.tolist()}
+                return {endpoint_name: predictions}
 
     def run(self, port: int = 8000, host: str = "127.0.0.1", **kw):
         """