Fix new 4

aimclub · Jul 4, 2024 · 9564e64 · 9564e64
1 parent 1942689
commit 9564e64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/integration/api/test_main_api.py b/test/integration/api/test_main_api.py
@@ -36,10 +36,10 @@ def test_api_predict_correct(task_type, metric_name):
     train_data, test_data, _ = get_dataset(task_type)
     changed_api_params = {
         **TESTS_MAIN_API_DEFAULT_PARAMS,
-        'timeout': 1,
+        'timeout': 2,
         'preset': 'fast_train'
     }
-    model = Fedot(problem=task_type, metric=metric_name, **changed_api_params)
+    model = Fedot(problem=task_type, metric=metric_name, **changed_api_params, cv_folds=2)
     fedot_model = model.fit(features=train_data)
     prediction = model.predict(features=test_data)
     metric = model.get_metrics(metric_names=metric_name, rounding_order=5)