From ed10c89433500a542a780dad5c8638ade45b910f Mon Sep 17 00:00:00 2001 From: Samhita Alla Date: Thu, 29 Feb 2024 10:31:29 +0530 Subject: [PATCH] upload files/directories when using remote-workflow (#2222) * enable remote Signed-off-by: Samhita Alla * modify flyte context Signed-off-by: Samhita Alla * remove dict update Signed-off-by: Samhita Alla * revert format Signed-off-by: Samhita Alla * revert format Signed-off-by: Samhita Alla --------- Signed-off-by: Samhita Alla Co-authored-by: Kevin Su Signed-off-by: Jan Fiedler --- flytekit/clis/sdk_in_container/run.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/flytekit/clis/sdk_in_container/run.py b/flytekit/clis/sdk_in_container/run.py index bd87f7b64b..e2887ce51a 100644 --- a/flytekit/clis/sdk_in_container/run.py +++ b/flytekit/clis/sdk_in_container/run.py @@ -595,7 +595,7 @@ def _get_params( defaults: typing.Optional[typing.Dict[str, Parameter]] = None, ) -> typing.List["click.Parameter"]: params = [] - flyte_ctx = context_manager.FlyteContextManager.current_context() + flyte_ctx = ctx.obj.remote_instance().context for name, var in inputs.items(): if fixed and name in fixed: continue @@ -609,6 +609,7 @@ def _get_params( return params def get_params(self, ctx: click.Context) -> typing.List["click.Parameter"]: + ctx.obj.remote = True if not self.params: self.params = [] entity = self._fetch_entity(ctx)