feat: make quickstart batchable (#3172)

Co-authored-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Nov 2, 2022 · 0e8b7b8 · 0e8b7b8
1 parent b67bd20
commit 0e8b7b8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/examples/quickstart/service.py b/examples/quickstart/service.py
@@ -9,7 +9,9 @@
 
 
 @svc.api(
-    input=NumpyNdarray.from_sample(np.array([[4.9, 3.0, 1.4, 0.2]], dtype=np.double)),
+    input=NumpyNdarray.from_sample(
+        np.array([[4.9, 3.0, 1.4, 0.2]], dtype=np.double), enforce_shape=False
+    ),
     output=NumpyNdarray(),
 )
 async def classify(input_series: np.ndarray) -> np.ndarray:

diff --git a/examples/quickstart/train.py b/examples/quickstart/train.py
@@ -18,5 +18,7 @@
     clf.fit(X, y)
 
     # Save model to BentoML local model store
-    saved_model = bentoml.sklearn.save_model("iris_clf", clf)
+    saved_model = bentoml.sklearn.save_model(
+        "iris_clf", clf, signatures={"predict": {"batchable": True, "batch_dim": 0}}
+    )
     print(f"Model saved: {saved_model}")