From 522b831cfa4564e2599f427fd625831486c19d34 Mon Sep 17 00:00:00 2001 From: Gabriel Tseng Date: Tue, 6 Jun 2023 17:34:13 +0200 Subject: [PATCH] Fix off by one error in inference.py --- openmapflow/inference.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openmapflow/inference.py b/openmapflow/inference.py index a13f79a3..e2405fdf 100644 --- 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)