Int test fix (#1355)

* relaxed time limits for test_early_stop_in_tuning
aimclub · Jan 7, 2025 · 16b0127 · 16b0127
1 parent 2c23695
commit 16b0127
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/integration/pipelines/tuning/test_pipeline_tuning.py b/test/integration/pipelines/tuning/test_pipeline_tuning.py
@@ -441,7 +441,7 @@ def test_early_stop_in_tuning(data_fixture, request):
                            loss_function=ClassificationMetricsEnum.ROCAUC,
                            iterations=1000,
                            early_stopping_rounds=1)
-    assert time() - start_pipeline_tuner < 1
+    assert time() - start_pipeline_tuner < 1.1
 
     start_sequential_tuner = time()
     _ = run_pipeline_tuner(tuner=SequentialTuner,
@@ -450,15 +450,15 @@ def test_early_stop_in_tuning(data_fixture, request):
                            loss_function=ClassificationMetricsEnum.ROCAUC,
                            iterations=1000,
                            early_stopping_rounds=1)
-    assert time() - start_sequential_tuner < 1
+    assert time() - start_sequential_tuner < 1.1
 
     start_node_tuner = time()
     _ = run_node_tuner(train_data=train_data,
                        pipeline=get_class_pipelines()[0],
                        loss_function=ClassificationMetricsEnum.ROCAUC,
                        iterations=1000,
                        early_stopping_rounds=1)
-    assert time() - start_node_tuner < 1
+    assert time() - start_node_tuner < 1.1
 
 
 def test_search_space_correctness_after_customization():