2222 ],
2323)
2424def test_with_regex (dataset , data_config , refit_after ):
25- pytest .importorskip (
26- "sentence_transformers" , reason = "Sentence Transformers library is required for regex-based pipelines"
27- )
2825 project_dir = setup_environment ()
2926 search_space = get_search_space ("regex" )
3027
@@ -37,10 +34,6 @@ def test_with_regex(dataset, data_config, refit_after):
3734
3835
3936def test_no_node_separation (dataset_no_oos ):
40- pytest .importorskip (
41- "sentence_transformers" , reason = "Sentence Transformers library is required for regex-based pipelines"
42- )
43-
4437 project_dir = setup_environment ()
4538 search_space = get_search_space ("light" )
4639
@@ -53,10 +46,6 @@ def test_no_node_separation(dataset_no_oos):
5346
5447
5548def test_full_config (dataset_no_oos ):
56- pytest .importorskip (
57- "sentence_transformers" , reason = "Sentence Transformers library is required for regex-based pipelines"
58- )
59-
6049 config_path = ires .files ("tests.assets.configs" ).joinpath ("full_training.yaml" )
6150 pipeline_optimizer = Pipeline .from_optimization_config (config_path )
6251 pipeline_optimizer .fit (dataset_no_oos , refit_after = False )
@@ -67,10 +56,6 @@ def test_full_config(dataset_no_oos):
6756 ["tpe" , "random" ],
6857)
6958def test_bayes (dataset , sampler ):
70- pytest .importorskip (
71- "sentence_transformers" , reason = "Sentence Transformers library is required for regex-based pipelines"
72- )
73-
7459 project_dir = setup_environment ()
7560 search_space = get_search_space ("optuna" )
7661
@@ -99,7 +84,6 @@ def test_bayes(dataset, sampler):
9984 ],
10085)
10186def test_cv (dataset , task_type ):
102- pytest .importorskip ("peft" )
10387 project_dir = setup_environment ()
10488 search_space = get_search_space (task_type )
10589
@@ -133,8 +117,6 @@ def test_cv(dataset, task_type):
133117 ],
134118)
135119def test_no_context_optimization (dataset , task_type ):
136- pytest .importorskip ("peft" )
137-
138120 project_dir = setup_environment ()
139121 search_space = get_search_space (task_type )
140122
@@ -166,8 +148,6 @@ def test_no_context_optimization(dataset, task_type):
166148 ],
167149)
168150def test_dump_modules (dataset , task_type ):
169- pytest .importorskip ("peft" )
170-
171151 project_dir = setup_environment ()
172152 search_space = get_search_space (task_type )
173153
0 commit comments