2222 ],
2323)
2424def test_with_regex (dataset , data_config , refit_after ):
25+ pytest .importorskip ("sentence_transformers" , reason = "Sentence Transformers library is required for regex-based pipelines" )
2526 project_dir = setup_environment ()
2627 search_space = get_search_space ("regex" )
2728
@@ -34,6 +35,8 @@ def test_with_regex(dataset, data_config, refit_after):
3435
3536
3637def test_no_node_separation (dataset_no_oos ):
38+ pytest .importorskip ("sentence_transformers" , reason = "Sentence Transformers library is required for regex-based pipelines" )
39+
3740 project_dir = setup_environment ()
3841 search_space = get_search_space ("light" )
3942
@@ -46,6 +49,8 @@ def test_no_node_separation(dataset_no_oos):
4649
4750
4851def test_full_config (dataset_no_oos ):
52+ pytest .importorskip ("sentence_transformers" , reason = "Sentence Transformers library is required for regex-based pipelines" )
53+
4954 config_path = ires .files ("tests.assets.configs" ).joinpath ("full_training.yaml" )
5055 pipeline_optimizer = Pipeline .from_optimization_config (config_path )
5156 pipeline_optimizer .fit (dataset_no_oos , refit_after = False )
@@ -56,6 +61,8 @@ def test_full_config(dataset_no_oos):
5661 ["tpe" , "random" ],
5762)
5863def test_bayes (dataset , sampler ):
64+ pytest .importorskip ("sentence_transformers" , reason = "Sentence Transformers library is required for regex-based pipelines" )
65+
5966 project_dir = setup_environment ()
6067 search_space = get_search_space ("optuna" )
6168
0 commit comments