Skip to content

Commit

Permalink
cli init aws tags option
Browse files Browse the repository at this point in the history
  • Loading branch information
RikishK committed Nov 18, 2024
1 parent bb237b8 commit dd7073a
Showing 1 changed file with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion metaflow/cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -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'
Expand All @@ -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,
Expand Down

0 comments on commit dd7073a

Please sign in to comment.