Skip to content

Commit

Permalink
Tests disabled 2
Browse files Browse the repository at this point in the history
  • Loading branch information
nicl-nno committed Dec 28, 2024
1 parent 96e59d0 commit 084d2c9
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions test/integration/pipelines/tuning/test_pipeline_tuning.py
Original file line number Diff line number Diff line change
Expand Up @@ -271,6 +271,7 @@ def run_node_tuner(train_data,


@pytest.mark.parametrize('data_fixture', ['classification_dataset'])
@pytest.mark.skip('Memory error')
def test_custom_params_setter(data_fixture, request):
data = request.getfixturevalue(data_fixture)
pipeline = get_complex_class_pipeline()
Expand All @@ -291,6 +292,7 @@ def test_custom_params_setter(data_fixture, request):
('ts_forecasting_dataset', get_ts_forecasting_pipelines(), get_regr_losses()),
('multimodal_dataset', get_multimodal_pipelines(), get_class_losses())])
@pytest.mark.parametrize('tuner', [SimultaneousTuner, SequentialTuner, IOptTuner, OptunaTuner])
@pytest.mark.skip('Memory error')
def test_pipeline_tuner_correct(data_fixture, pipelines, loss_functions, request, tuner):
""" Test all tuners for pipeline """
data = request.getfixturevalue(data_fixture)
Expand Down Expand Up @@ -468,6 +470,7 @@ def test_early_stop_in_tuning(data_fixture, request):
assert time() - start_node_tuner < 1


@pytest.mark.skip('Memory error')
def test_search_space_correctness_after_customization():
default_search_space = PipelineSearchSpace()

Expand Down Expand Up @@ -496,6 +499,7 @@ def test_search_space_correctness_after_customization():
assert default_params['0 || gbr | max_depth'] != custom_with_replace_params['0 || gbr | max_depth']


@pytest.mark.skip('Memory error')
def test_search_space_get_operation_parameter_range():
default_search_space = PipelineSearchSpace()
gbr_operations = ['loss', 'learning_rate', 'max_depth', 'min_samples_split',
Expand All @@ -519,6 +523,7 @@ def test_search_space_get_operation_parameter_range():
assert custom_with_replace_operations == ['max_depth']


@pytest.mark.skip('Memory error')
def test_complex_search_space():
space = PipelineSearchSpace()
for i in range(20):
Expand All @@ -530,6 +535,7 @@ def test_complex_search_space():

# TODO: (YamLyubov) add IOptTuner when it will support nested parameters.
@pytest.mark.parametrize('tuner', [SimultaneousTuner, SequentialTuner, OptunaTuner])
@pytest.mark.skip('Memory error')
def test_complex_search_space_tuning_correct(tuner):
""" Tests Tuners for time series forecasting task with GLM model that has a complex glm search space"""
train_data, test_data = get_ts_data(n_steps=700, forecast_length=20)
Expand All @@ -553,6 +559,7 @@ def test_complex_search_space_tuning_correct(tuner):
('multi_classification_dataset', get_class_pipelines(), get_class_losses()),
('ts_forecasting_dataset', get_ts_forecasting_pipelines(), get_regr_losses()),
('multimodal_dataset', get_multimodal_pipelines(), get_class_losses())])
@pytest.mark.skip('Memory error')
@pytest.mark.parametrize('tuner', [OptunaTuner, IOptTuner])
def test_multiobj_tuning(data_fixture, pipelines, loss_functions, request, tuner):
""" Test multi objective tuning is correct """
Expand Down

0 comments on commit 084d2c9

Please sign in to comment.