From ca6eca51d0652d696d2b55914887c87bf878ba45 Mon Sep 17 00:00:00 2001
From: Aaron Pham <29749331+aarnphm@users.noreply.github.com>
Date: Fri, 1 Sep 2023 12:08:40 -0400
Subject: [PATCH] fix(pull): correct use `cloud_context` for models pull
 (#4163)

Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
---
 src/bentoml_cli/models.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/bentoml_cli/models.py b/src/bentoml_cli/models.py
index 1e1fb585d34..365c74f57cd 100644
--- 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: