Skip to content

Commit 5072810

Browse files
committed
add test for random sampler
1 parent 2fb6ac0 commit 5072810

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/pipeline/test_optimization.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,11 @@
1111
from tests.conftest import get_search_space, setup_environment
1212

1313

14-
def test_bayes(dataset):
14+
@pytest.mark.parametrize(
15+
"tuning",
16+
["bayes", "random"],
17+
)
18+
def test_bayes(dataset, tuning):
1519
project_dir = setup_environment()
1620
search_space = get_search_space("bayes")
1721

@@ -21,7 +25,7 @@ def test_bayes(dataset):
2125
pipeline_optimizer.set_config(VectorIndexConfig())
2226
pipeline_optimizer.set_config(EmbedderConfig(batch_size=16, max_length=32, device="cpu"))
2327

24-
pipeline_optimizer.fit(dataset, scheme="cv", refit_after=True, tuning="bayes")
28+
pipeline_optimizer.fit(dataset, scheme="ho", refit_after=False, tuning=tuning)
2529

2630

2731
@pytest.mark.parametrize(

0 commit comments

Comments
 (0)