Fixes

aimclub · Aug 10, 2023 · 442fad8 · 442fad8
1 parent 5b1b8ce
commit 442fad8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/unit/data/test_data_split.py b/test/unit/data/test_data_split.py
@@ -312,7 +312,7 @@ def test_stratify(cv_folds, shuffle, stratify, data_classes):
                            get_tabular_classification_data(length=100, class_count=4)),  # cv_folds classification
                           (True, True, None,
                            get_tabular_classification_data(length=100, class_count=4)),  # holdout classification
-                          (False, True, 2, get_ts_data_to_forecast(10, 100)), 1 # cv_folds timeseries
+                          (False, True, 2, get_ts_data_to_forecast(10, 100)),  # cv_folds timeseries
                           (False, True, None, get_ts_data_to_forecast(10, 100)),  # holdout timeseries
                           ])
 def test_shuffle(is_shuffle, cv_folds, shuffle, data):