fix(serve): return self within context manager (#3504)

bentoml · Feb 2, 2023 · aded17c · aded17c
1 parent b1e95b0
commit aded17c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/bentoml/_internal/server/server.py b/src/bentoml/_internal/server/server.py
@@ -39,7 +39,7 @@ def address(self) -> str:
         return f"{self.host}:{self.port}"
 
     def __enter__(self):
-        yield self
+        return self
 
     def __exit__(
         self,