diff --git a/mlpp_lib/model_selection.py b/mlpp_lib/model_selection.py index a853499..52ff9f9 100644 --- a/mlpp_lib/model_selection.py +++ b/mlpp_lib/model_selection.py @@ -209,7 +209,7 @@ def _time_partitioning(self) -> None: end_date = np.datetime64(test_indexers[1]) test_indexers = self.time_index[ np.logical_and( - self.time_index >= start_date, self.time_index <= end_date + self.time_index >= start_date, self.time_index < end_date ) ] test_indexers = [t for t in test_indexers if t in self.time_index]