FIX: private method test case

tinkoff-ai · Apr 18, 2022 · 596f4b5 · 596f4b5
1 parent 1fb0b16
commit 596f4b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_pipeline/test_pipeline.py b/tests/test_pipeline/test_pipeline.py
@@ -451,7 +451,7 @@ def test_run_fold(ts_run_fold: TSDataset, mask: FoldMask, expected: Dict[str, Li
     )
 
     pipeline = Pipeline(model=NaiveModel(lag=5), transforms=[], horizon=4)
-    fold = pipeline._run_fold(train, test, 1, mask, [MAE()])
+    fold = pipeline._run_fold(train, test, 1, mask, [MAE()], forecast_params=dict())
     for seg in fold["metrics"]["MAE"].keys():
         assert fold["metrics"]["MAE"][seg] == expected[seg]