From dd7073ae03d749965681c975a0232c4cf0500a9b Mon Sep 17 00:00:00 2001 From: RikishK Date: Tue, 19 Nov 2024 10:50:06 +1100 Subject: [PATCH] cli init aws tags option --- metaflow/cli.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/metaflow/cli.py b/metaflow/cli.py index 9af2678fd1c..e732b43d339 100644 --- a/metaflow/cli.py +++ b/metaflow/cli.py @@ -504,8 +504,13 @@ def step( default=None, help="Tags for this instance of the step.", ) +@click.option( + "--aws-tags", + multiple=True, + default=None, + help="AWS tags.") @click.pass_obj -def init(obj, run_id=None, task_id=None, tags=None, **kwargs): +def init(obj, run_id=None, task_id=None, tags=None, aws_tags=None, **kwargs): # init is a separate command instead of an option in 'step' # since we need to capture user-specified parameters with # @add_custom_parameters. Adding custom parameters to 'step' @@ -515,6 +520,7 @@ def init(obj, run_id=None, task_id=None, tags=None, **kwargs): # variables. obj.metadata.add_sticky_tags(tags=tags) + obj.metadata.add_sticky_tags(tags=aws_tags) runtime = NativeRuntime( obj.flow,