fix bundle index bug when specmin>0

desihub · Jan 31, 2021 · 6f96da9 · 6f96da9
1 parent db50416
commit 6f96da9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/py/desispec/scripts/specex.py b/py/desispec/scripts/specex.py
@@ -149,13 +149,12 @@ def main(args, comm=None):
         rank = comm.rank
 
     mynbundle = int(nbundle / nproc)
-    myfirstbundle = 0
     leftover = nbundle % nproc
     if rank < leftover:
         mynbundle += 1
-        myfirstbundle = rank * mynbundle
+        myfirstbundle = bundles[0] + rank * mynbundle
     else:
-        myfirstbundle = ((mynbundle + 1) * leftover) + \
+        myfirstbundle = bundles[0] + ((mynbundle + 1) * leftover) + \
             (mynbundle * (rank - leftover))
 
     if rank == 0: