From 6f96da901dd96ace47eb2609562bf9fcee94eb1a Mon Sep 17 00:00:00 2001 From: Stephen Bailey Date: Sun, 31 Jan 2021 13:31:34 -0800 Subject: [PATCH] fix bundle index bug when specmin>0 --- py/desispec/scripts/specex.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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: