f

dream-faster · Feb 29, 2024 · b2af25a · b2af25a
1 parent 6530ef0
commit b2af25a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/wrappers/test_lightgbm.py b/tests/wrappers/test_lightgbm.py
@@ -1,6 +1,6 @@
 from fold.loop import train_backtest
 from fold.models.wrappers.gbd import WrapLGBM
-from fold.splitters import ExpandingWindowSplitter, SingleWindowSplitter
+from fold.splitters import ExpandingWindowSplitter, ForwardSingleWindowSplitter
 from fold.utils.tests import (
     generate_monotonous_data,
     generate_sine_wave_data,
@@ -63,7 +63,7 @@ def test_automatic_wrapping_lgbm() -> None:
         LGBMRegressor(),
         X,
         y,
-        splitter=SingleWindowSplitter(0.5),
+        splitter=ForwardSingleWindowSplitter(0.5),
     )