upd test

Author: voorhs

tests/callback/test_callback.py

@@ -5,7 +5,7 @@
 
 from autointent import Context, Pipeline
 from autointent._callbacks import CallbackHandler, OptimizerCallback
-from autointent.configs import LoggingConfig, VectorIndexConfig
+from autointent.configs import DataConfig, LoggingConfig, VectorIndexConfig
 from tests.conftest import setup_environment
 
 
@@ -86,7 +86,7 @@ def test_pipeline_callbacks(dataset):
     context.configure_vector_index(VectorIndexConfig(save_db=True))
     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)
+    context.set_dataset(dataset, DataConfig(scheme="ho", separate_nodes=True))
 
     pipeline_optimizer._fit(context)