fix CLI

EpistasisLab · Jun 28, 2017 · 8672442 · 8672442
1 parent 8a20919
commit 8672442
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tpot/driver.py b/tpot/driver.py
@@ -451,8 +451,11 @@ def load_scoring_function(scoring_func):
 
     return scoring_func
 
-def main(args):
+def main(args=None):
     """Perform a TPOT run."""
+    if args is None:
+        args = sys.argv[1:]
+
     if args.VERBOSITY >= 2:
         _print_args(args)