Merge pull request #107 from arbennett/develop

Fix last gru index for distributed
UW-Hydro · May 18, 2020 · 5f55658 · 5f55658
2 parents 7f781c3 + 851775a
commit 5f55658
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysumma/distributed.py b/pysumma/distributed.py
@@ -115,7 +115,7 @@ def _generate_args(self, chunk_size: int=None, num_chunks: int=None):
         if chunk_size:
             sim_truncated = (chunk_size-1) * (sim_size // (chunk_size-1))
             starts = np.arange(1, sim_truncated+1, chunk_size).astype(int)
-            stops = np.append(starts[1:], sim_size)
+            stops = np.append(starts[1:], sim_size+1)
             chunks = np.vstack([starts, stops]).T
         elif num_chunks:
             chunk_size = np.ceil(sim_size / num_chunks).astype(int)