pyflyte run use config path from arg value

Signed-off-by: Kevin Su <[email protected]>
flyteorg · May 11, 2022 · 642003c · 642003c
1 parent c011ef7
commit 642003c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/flytekit/clis/sdk_in_container/run.py b/flytekit/clis/sdk_in_container/run.py
@@ -13,6 +13,7 @@
 from pytimeparse import parse
 
 from flytekit import BlobType, Literal, Scalar
+from flytekit.clis.sdk_in_container.constants import CTX_CONFIG_FILE, CTX_DOMAIN, CTX_PROJECT
 from flytekit.configuration import Config, ImageConfig, SerializationSettings
 from flytekit.configuration.default_images import DefaultImages
 from flytekit.core import context_manager, tracker
@@ -527,9 +528,9 @@ def get_command(self, ctx, workflow):
         wf_entity = load_naive_entity(module, workflow)
 
         # If this is a remote execution, which we should know at this point, then create the remote object
-        p = ctx.obj[RUN_LEVEL_PARAMS_KEY].get("project")
-        d = ctx.obj[RUN_LEVEL_PARAMS_KEY].get("domain")
-        r = FlyteRemote(Config.auto(), default_project=p, default_domain=d)
+        p = ctx.obj[RUN_LEVEL_PARAMS_KEY].get(CTX_PROJECT)
+        d = ctx.obj[RUN_LEVEL_PARAMS_KEY].get(CTX_DOMAIN)
+        r = FlyteRemote(Config.auto(ctx.obj.get(CTX_CONFIG_FILE)), default_project=p, default_domain=d)
         ctx.obj[FLYTE_REMOTE_INSTANCE_KEY] = r
         get_upload_url_fn = functools.partial(r.client.get_upload_signed_url, project=p, domain=d)