diff --git a/sklearn_questions.py b/sklearn_questions.py index a0b9d8e..1890fa7 100644 --- a/sklearn_questions.py +++ b/sklearn_questions.py @@ -205,7 +205,6 @@ def split(self, X, y, groups=None): idx_test : ndarray The testing set indices for that split. """ - if self.time_col != 'index': X = X.set_index(self.time_col)