bug fix

Author: voorhs

autointent/custom_types/_types.py

@@ -115,11 +115,5 @@ class Split:
 - `filter`: drop incompatible modules from search space
 """
 
-SearchSpacePreset = Literal["light", "light_extra", "heavy", "heavy_moderate"]
-"""Some presets that our library supports from lightest to heaviest:
-
-- `light_extra`: just like `light` but tuned with :py:class:`optuna.samplers.RandomSampler`
-- `light`: only light-weight modules tuned with :py:class:`optuna.samplers.TPESampler`
-- `heavy_moderate`: just like `heavy` but tuned with :py:class:`optuna.samplers.RandomSampler`
-- `light`: light-weight modules accompanied with heavy-weight ones tuned with :py:class:`optuna.samplers.TPESampler`
-"""
+SearchSpacePreset = Literal["light", "heavy"]
+"""Some presets that our library supports from lightest to heaviest"""

tests/callback/test_callback.py

@@ -5,7 +5,7 @@
 
 from autointent import Context, Pipeline
 from autointent._callbacks import CallbackHandler, OptimizerCallback
-from autointent.configs import DataConfig, LoggingConfig
+from autointent.configs import DataConfig, HPOConfig, LoggingConfig
 from tests.conftest import setup_environment
 
 
@@ -87,8 +87,9 @@ def test_pipeline_callbacks(dataset):
     context.configure_logging(LoggingConfig(run_name="dummy_run_name", project_dir=project_dir, dump_modules=False))
     context.callback_handler = CallbackHandler([DummyCallback])
     context.set_dataset(dataset, DataConfig(scheme="ho"))
+    context.configure_hpo(HPOConfig(n_trials=10))
 
-    pipeline_optimizer._fit(context, "tpe")
+    pipeline_optimizer._fit(context)
 
     dummy_callback = context.callback_handler.callbacks[0]