diff --git a/py/desispec/scripts/specex.py b/py/desispec/scripts/specex.py index a64fb2c21..d354d6a17 100644 --- 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: