UP my solution

x-datascience-datacamp · Dec 22, 2023 · 41da450 · 41da450
1 parent 2cc9394
commit 41da450
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/sklearn_questions.py b/sklearn_questions.py
@@ -182,12 +182,11 @@ def get_n_splits(self, X, y=None, groups=None):
         n_splits : int
             The number of splits.
         """
-
         X = X.reset_index()
         self.time_ = X.loc[:, self.time_col_]
 
         if self.time_.dtype != 'datetime64[ns]':
-            raise ValueError('not a datetime')
+            raise ValueError('Chosen column is not a datetime')
         self.months_ = (np.sort((self.time_.dt.strftime('%Y-%m')).unique()))
         print(self.months_)