diff --git a/metaflow/plugins/aws/eks/kubernetes_decorator.py b/metaflow/plugins/aws/eks/kubernetes_decorator.py index 813aab054a4..2f06b5fe0fc 100644 --- a/metaflow/plugins/aws/eks/kubernetes_decorator.py +++ b/metaflow/plugins/aws/eks/kubernetes_decorator.py @@ -187,7 +187,7 @@ def runtime_step_cli( if k == 'namespace': cli_args.command_options['k8s_namespace'] = v else: - cli_args.command_options['namespace'] = v + cli_args.command_options[k] = v cli_args.command_options["run-time-limit"] = self.run_time_limit cli_args.entrypoint[0] = sys.executable