Add --squeeze to the parent argparser. (#20)

vanvalenlab · Aug 27, 2021 · 8eaf6ff · 8eaf6ff
1 parent e9bddee
commit 8eaf6ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_app.py b/run_app.py
@@ -80,7 +80,7 @@ def existing_file(x):
                         choices=('DEBUG', 'INFO', 'WARN', 'ERROR', 'CRITICAL'),
                         help='Only log the given level and above.')
 
-    parser.add_argument('--squeeze', action='store_true',
+    parent.add_argument('--squeeze', action='store_true',
                         help='Squeeze the output tensor before saving.')
 
     # use subparsers to group options for different applications