change None to 'none' bc Choice must be a string

Netflix · May 8, 2024 · a6d5c63 · a6d5c63
1 parent d4cf28a
commit a6d5c63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metaflow/plugins/aws/batch/batch_cli.py b/metaflow/plugins/aws/batch/batch_cli.py
@@ -151,7 +151,7 @@ def kill(ctx, run_id, user, my_runs):
 @click.option("--tmpfs-size", help="tmpfs requirement for AWS Batch.")
 @click.option("--tmpfs-path", help="tmpfs requirement for AWS Batch.")
 # TODO: Maybe remove it altogether since it's not used here
-@click.option("--ubf-context", default=None, type=click.Choice([None, "ubf_control"]))
+@click.option("--ubf-context", default=None, type=click.Choice(["none", "ubf_control"]))
 @click.option("--host-volumes", multiple=True)
 @click.option("--efs-volumes", multiple=True)
 @click.option(