return string only

bentoml · Nov 13, 2024 · e3ac420 · e3ac420
1 parent 69b0d84
commit e3ac420
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/e2e/bento_new_sdk/test_asgi.py b/tests/e2e/bento_new_sdk/test_asgi.py
@@ -10,16 +10,16 @@ class Model:
     model = {"hello": "world"}
 
     @bentoml.api()
-    def get(self, key: str) -> str | None:
-        return self.model.get(key)
+    def get(self, key: str) -> str:
+        return self.model.get(key, "")
 
 
 @bentoml.service(metrics={"enabled": False})
 class Service:
     model = bentoml.depends(Model)
 
     @bentoml.api()
-    def get(self, key: str) -> dict[str, str | None]:
+    def get(self, key: str) -> dict[str, str]:
         return {"value": self.model.get(key)}