From c7248a2d236be0fa1521649769ec854595c1174f Mon Sep 17 00:00:00 2001 From: Aaron Pham <29749331+aarnphm@users.noreply.github.com> Date: Mon, 31 Oct 2022 13:00:05 -0700 Subject: [PATCH] fix: pandas openAPI handle nulltype Signed-off-by: Aaron Pham <29749331+aarnphm@users.noreply.github.com> --- src/bentoml/_internal/io_descriptors/pandas.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/bentoml/_internal/io_descriptors/pandas.py b/src/bentoml/_internal/io_descriptors/pandas.py index 84676b501d..90e44ec69a 100644 --- a/src/bentoml/_internal/io_descriptors/pandas.py +++ b/src/bentoml/_internal/io_descriptors/pandas.py @@ -356,6 +356,8 @@ def _convert_dtype( return str(value) elif isinstance(value, dict): return {str(k): self._convert_dtype(v) for k, v in value.items()} + elif value is None: + return "null" else: logger.warning(f"{type(value)} is not yet supported.") return None @@ -822,6 +824,8 @@ def _convert_dtype( return str(value) elif isinstance(value, dict): return {str(k): self._convert_dtype(v) for k, v in value.items()} + elif value is None: + return "null" else: logger.warning(f"{type(value)} is not yet supported.") return None