allow mpi mode with files_from

segasai · Nov 21, 2024 · a3b75a5 · a3b75a5
1 parent c5d9e32
commit a3b75a5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/py/rvspecfit/desi/desi_fit.py b/py/rvspecfit/desi/desi_fit.py
@@ -1811,6 +1811,9 @@ def main(args):
                           file_from=input_file_from,
                           queue=queue_file)
     else:
+        if input_files is None:
+            with open(input_file_from, 'r') as fp:
+                input_files = [_.rstrip() for _ in fp.readlines()]
         files = utils.MPIFileQueue(file_list=input_files)
 
     if (not args.mpi) or (args.mpi and rank != 0):