From a1a91cb412be8d39afe50b5d8970804595158bdd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=92=D0=B0=D0=B4=D0=B8=D0=BC=20=D0=90=D1=85=D0=BC=D0=B5?= =?UTF-8?q?=D1=80=D0=BE=D0=B2?= Date: Tue, 7 Jan 2025 00:17:05 +0500 Subject: [PATCH] relaxed time limits for test_early_stop_in_tuning --- test/integration/pipelines/tuning/test_pipeline_tuning.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/integration/pipelines/tuning/test_pipeline_tuning.py b/test/integration/pipelines/tuning/test_pipeline_tuning.py index 0d052c0c75..cfd0c3787b 100644 --- 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,7 +450,7 @@ 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, @@ -458,7 +458,7 @@ def test_early_stop_in_tuning(data_fixture, request): 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():