2.7 compatibility argparse

datarobot · May 12, 2016 · 512c929 · 512c929
1 parent 12281dc
commit 512c929
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/datarobot_batch_scoring/main.py b/datarobot_batch_scoring/main.py
@@ -166,8 +166,10 @@ def main(argv=sys.argv[1:]):
         defaults.update(file_args)
     parser.set_defaults(**defaults)
     for action in parser._actions:
-        if action.dest in defaults:
+        if action.dest in defaults and action.required:
             action.required = False
+            if '--' + action.dest not in argv:
+                action.nargs = '?'
     parsed_args = {k: v
                    for k, v in vars(parser.parse_args(argv)).items()
                    if v is not None}