fix(pull): correct use cloud_context for models pull (#4163)

Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
bentoml · Sep 1, 2023 · ca6eca5 · ca6eca5
1 parent d642016
commit ca6eca5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/bentoml_cli/models.py b/src/bentoml_cli/models.py
@@ -268,7 +268,9 @@ def pull(ctx: click.Context, model_tag: str | None, force: bool, bentofile: str)
         if model_tag is not None:
             if ctx.get_parameter_source("bentofile") != ParameterSource.DEFAULT:
                 click.echo("-f bentofile is ignored when model_tag is provided")
-            cloud_client.pull_model(model_tag, force=force, context=ctx.obj.context)
+            cloud_client.pull_model(
+                model_tag, force=force, context=ctx.obj.cloud_context
+            )
             return
 
         try: