From d5247382879a07d2aca3bb622579b7341abf2dbb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Louis=20Poulain--Auz=C3=A9au?= <47986600+louisPoulain@users.noreply.github.com> Date: Fri, 8 Nov 2024 13:59:07 +0100 Subject: [PATCH] Use strict inequality for more clarity --- mlpp_lib/model_selection.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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]