fix: types

Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
bentoml · Nov 8, 2022 · 629f363 · 629f363
1 parent 591418b
commit 629f363
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/bentoml/_internal/service/openapi/specification.py b/src/bentoml/_internal/service/openapi/specification.py
@@ -104,7 +104,7 @@ class Schema:
     anyOf: t.Optional[t.List[Schema]] = None
     not_: t.Optional[Schema] = None
     items: t.Optional[t.Union[Schema, t.List[Schema]]] = None
-    properties: t.Optional[t.Dict[str, Schema | Reference]] = None
+    properties: t.Optional[t.Dict[str, t.Union[Schema, Reference]]] = None
     additionalProperties: t.Optional[t.Union[Schema, Reference, bool]] = None
     description: t.Optional[str] = None
     format: t.Optional[str] = None

diff --git a/tests/unit/_internal/io/test_numpy.py b/tests/unit/_internal/io/test_numpy.py
@@ -85,6 +85,8 @@ def test_numpy_openapi_request_body():
 
     nparray = NumpyNdarray(dtype="float")
     nparray.sample_input = ExampleGeneric("asdf")  # type: ignore (test exception)
+    with pytest.raises(BadInput):
+        nparray.openapi_example()
 
 
 def test_numpy_openapi_responses():