fix imports

bentoml · Oct 10, 2023 · 2b20a88 · 2b20a88
1 parent 646d42d
commit 2b20a88
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/bentoml/_internal/client/__init__.py b/src/bentoml/_internal/client/__init__.py
@@ -84,7 +84,7 @@ def wait_until_server_ready(
     @staticmethod
     def from_url(
         server_url: str, *, kind: None | t.Literal["auto"] = ...
-    ) -> GrpcClient | HTTPClient:
+    ) -> Client:
         ...
 
     @t.overload

diff --git a/src/bentoml/server.py b/src/bentoml/server.py
@@ -17,8 +17,8 @@
 
 from ._internal.bento import Bento
 from ._internal.client import Client
-from ._internal.client import GrpcClient
-from ._internal.client import HTTPClient
+from ._internal.client.grpc import GrpcClient
+from ._internal.client.http import HTTPClient
 from ._internal.configuration.containers import BentoMLContainer
 from ._internal.service import Service
 from ._internal.tag import Tag