upd test

Author: voorhs

tests/pipeline/test_optimization.py

@@ -4,7 +4,7 @@
 import pytest
 
 from autointent import Pipeline
-from autointent.configs import DataConfig, LoggingConfig
+from autointent.configs import DataConfig, HPOConfig, LoggingConfig
 from tests.conftest import get_search_space, setup_environment
 
 
@@ -63,8 +63,9 @@ def test_bayes(dataset, sampler):
 
     pipeline_optimizer.set_config(LoggingConfig(project_dir=project_dir, dump_modules=True, clear_ram=True))
     pipeline_optimizer.set_config(DataConfig(scheme="ho", separation_ratio=0.5))
+    pipeline_optimizer.set_config(HPOConfig(sampler=sampler))
 
-    pipeline_optimizer.fit(dataset, refit_after=False, sampler=sampler)
+    pipeline_optimizer.fit(dataset, refit_after=False)
 
 
 @pytest.mark.parametrize(