diff --git a/src/wela/growth_rate.py b/src/wela/growth_rate.py index e95fbf8cf6862349d0cf66fef011c460eb563612..b9bef1e02477a597ea0142ae0bd15f39d4eb99c0 100644 --- a/src/wela/growth_rate.py +++ b/src/wela/growth_rate.py @@ -238,7 +238,11 @@ def run_parallel(t, volume, buddings_df, buddings, no_cells, config): t_batch = [t for i in range(batch_size)] for idx in range(0, no_cells, batch_size): print(f" mother {idx:d}/{no_cells-1:d}") - v_batch = [volume[idx + i, :] for i in range(batch_size)] + if idx + batch_size < no_cells: + v_batch = [volume[idx + i, :] for i in range(batch_size)] + else: + v_batch = [volume[i, :] for i in range(idx, no_cells)] + t_batch = [t for i in range(len(v_batch))] res_batch = pool.map(batch_GP, t_batch, v_batch) for ires in res_batch: if ires: