diff --git a/sklearn_questions.py b/sklearn_questions.py index 7f7fa5a..e483cac 100644 --- a/sklearn_questions.py +++ b/sklearn_questions.py @@ -203,7 +203,6 @@ def split(self, X, y, groups=None): idx_test : ndarray The testing set indices for that split. """ - if (X.reset_index().loc[:, self.time_col].dtype != "datetime64[ns]"): raise ValueError(f"{self.time_col} must be of type datetime.")