Merge pull request #131 from nasaharvest/inference-fix

Fix off by one error in inference.py
nasaharvest · Aug 31, 2023 · 144723e · 144723e
2 parents 773ef84 + 522b831
commit 144723e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openmapflow/inference.py b/openmapflow/inference.py
@@ -83,7 +83,7 @@ def run(
 
         batches = [
             x_np[i : i + self.batch_size]  # noqa: E203
-            for i in range(0, (x_np.shape[0] - 1), self.batch_size)
+            for i in range(0, x_np.shape[0], self.batch_size)
         ]
         batch_predictions = [self._on_single_batch(b) for b in batches]
         combined_pred = self._combine_predictions(flat_lat, flat_lon, batch_predictions)