From dd63f46e5ebbd445d6f75c61ba29c580ae6b2471 Mon Sep 17 00:00:00 2001 From: Aaron Pham <29749331+aarnphm@users.noreply.github.com> Date: Tue, 25 Oct 2022 18:33:44 -0700 Subject: [PATCH] fix: sample shouldn't be a memoryview [skip ci] depends on #3144 to be merged. Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com> --- src/bentoml/_internal/io_descriptors/numpy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/bentoml/_internal/io_descriptors/numpy.py b/src/bentoml/_internal/io_descriptors/numpy.py index c9c63984298..bac076e4c4e 100644 --- a/src/bentoml/_internal/io_descriptors/numpy.py +++ b/src/bentoml/_internal/io_descriptors/numpy.py @@ -284,7 +284,7 @@ def openapi_example(self) -> t.Any: if isinstance(self.sample, np.generic): raise BadInput("NumpyNdarray: sample must be a numpy array.") from None # NOTE: we only need to - return self.sample.ravel().tolist() + return self.sample.tolist() def openapi_request_body(self) -> dict[str, t.Any]: return {