Fix distribution inputs for restart

jaisw7 · Sep 19, 2019 · 35ca276 · 35ca276
1 parent b4fc168
commit 35ca276
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dgfs1D/initialize.py b/dgfs1D/initialize.py
@@ -20,7 +20,8 @@ def initialize():
     # Restart command
     ap_restart = sp.add_parser('restart', help='restart --help')
     ap_restart.add_argument('inp', type=FileType('r'), help='input file')
-    ap_restart.add_argument('dist', type=FileType('r'), help='distribution')
+    ap_restart.add_argument('dist', nargs='+', action='append', 
+        type=FileType('r'), default=[], help='sorted list of distributions')
     ap_restart.add_argument('-v', nargs=2, action='append', default=[], 
         help='substitute variables. Example: -v mesh::Ne 4')
     ap_restart.set_defaults(process_restart=True)