File tree Expand file tree Collapse file tree 1 file changed +2
-5
lines changed
Expand file tree Collapse file tree 1 file changed +2
-5
lines changed Original file line number Diff line number Diff line change 11import pytest
22
33from autointent import Context , Dataset
4- from autointent .configs import (
5- LoggingConfig ,
6- VectorIndexConfig ,
7- )
4+ from autointent .configs import DataConfig , LoggingConfig , VectorIndexConfig
85from autointent .nodes import NodeOptimizer
96from tests .conftest import get_dataset_path , setup_environment
107
@@ -79,7 +76,7 @@ def get_context(multilabel):
7976 dataset = Dataset .from_json (get_dataset_path ())
8077 if multilabel :
8178 dataset = dataset .to_multilabel ()
82- res .set_dataset (dataset )
79+ res .set_dataset (dataset , DataConfig ( scheme = "ho" , separate_nodes = True ) )
8380 res .configure_logging (LoggingConfig (project_dir = project_dir , dump_modules = True ))
8481 res .configure_vector_index (VectorIndexConfig ())
8582 return res
You can’t perform that action at this time.
0 commit comments