Skip to content

Commit 000f017

Browse files
committed
expect a list of prototypes if checking prototype
1 parent 25baff1 commit 000f017

2 files changed

Lines changed: 2 additions & 5 deletions

File tree

vetiver/handlers/base.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,3 @@
1-
from typing import List
21
from functools import singledispatch
32
from contextlib import suppress
43
import pandas as pd
@@ -191,7 +190,7 @@ def _prepare_data(pred_data) -> pd.DataFrame:
191190

192191

193192
@_prepare_data.register(pydantic.BaseModel)
194-
@_prepare_data.register(List)
193+
@_prepare_data.register(list)
195194
def _basemodel_list_data(pred_data):
196195

197196
return pd.DataFrame([dict(s) for s in pred_data])

vetiver/server.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -255,9 +255,7 @@ def predict(endpoint, data: Union[dict, pd.DataFrame, pd.Series], **kw) -> pd.Da
255255
if isinstance(data, pd.DataFrame):
256256
response = requester.post(endpoint, data=data.to_json(orient="records"), **kw)
257257
elif isinstance(data, pd.Series):
258-
response = requester.post(endpoint, data=data.to_json(), **kw)
259-
elif isinstance(data, list):
260-
response = requester.post(endpoint, json=data, **kw)
258+
response = requester.post(endpoint, json=[data.to_dict()], **kw)
261259
elif isinstance(data, dict):
262260
response = requester.post(endpoint, json=[data], **kw)
263261
else:

0 commit comments

Comments
 (0)