fix: do not initialize cloud client when building bento (#4998)

Signed-off-by: Frost Ming <[email protected]>
bentoml · Sep 25, 2024 · 7cc0d0e · 7cc0d0e
1 parent 27267ec
commit 7cc0d0e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/bentoml_cli/bentos.py b/src/bentoml_cli/bentos.py
@@ -395,7 +395,6 @@ def push(
         help="Platform to build for",
         type=click.Choice(ALLOWED_PLATFORMS),
     )
-    @inject
     def build(  # type: ignore (not accessed)
         build_ctx: str,
         bentofile: str,
@@ -407,7 +406,6 @@ def build(  # type: ignore (not accessed)
         threads: int,
         containerize: bool,
         platform: str | None,
-        _cloud_client: BentoCloudClient = Provide[BentoMLContainer.bentocloud_client],
     ):
         """Build a new Bento from current directory."""
         from bentoml._internal.configuration import set_quiet_mode
@@ -462,7 +460,8 @@ def build(  # type: ignore (not accessed)
         if push:
             if not get_quiet_mode():
                 rich.print(f"\n[magenta]Pushing {bento} to BentoCloud...[/]")
-            _cloud_client.bento.push(bento, force=force, threads=threads)
+            cloud_client = BentoMLContainer.bentocloud_client.get()
+            cloud_client.bento.push(bento, force=force, threads=threads)
         elif containerize:
             backend: DefaultBuilder = t.cast(
                 "DefaultBuilder", os.getenv("BENTOML_CONTAINERIZE_BACKEND", "docker")