summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--cli/app/commands/biggan/search.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/cli/app/commands/biggan/search.py b/cli/app/commands/biggan/search.py
index ee7167e..76e95ff 100644
--- a/cli/app/commands/biggan/search.py
+++ b/cli/app/commands/biggan/search.py
@@ -33,7 +33,7 @@ from app.search.vector import truncated_z_sample, truncated_z_single, create_lab
help='Tag this dataset')
# @click.option('-r', '--recursive', 'opt_recursive', is_flag=True)
@click.pass_context
-def cli(ctx, opt_fp_in, opt_dims, opt_video):
+def cli(ctx, opt_fp_in, opt_dims, opt_video, opt_tag):
"""
Search for an image (class vector) in BigGAN using gradient descent
"""