Hotfix #1021 after merging into master (#1032)

Hotfix for #1021 - Changed preset from auto to glm and increase tuning timeout for failed test
aimclub · Jan 31, 2023 · 993fcb9 · 993fcb9
1 parent aa4fdd3
commit 993fcb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/unit/api/test_main_api.py b/test/unit/api/test_main_api.py
@@ -166,11 +166,11 @@ def test_api_predict_correct(task_type, predefined_model, metric_name):
 @pytest.mark.parametrize('task_type, metric_name, pred_model', [
     ('classification', 'f1', 'dt'),
     ('regression', 'rmse', 'dtreg'),
-    ('ts_forecasting', 'rmse', 'auto')
+    ('ts_forecasting', 'rmse', 'glm')
 ])
 def test_api_tune_correct(task_type, metric_name, pred_model):
     if task_type is 'ts_forecasting':
-        tuning_timeout = 0.5
+        tuning_timeout = 0.25
         forecast_length = 5
         train_data, test_data, _ = get_dataset(task_type, validation_blocks=1)
         model = Fedot(