Skip to content

Commit dd5014c

Browse files
committed
upd test
1 parent 7890fba commit dd5014c

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

tests/nodes/conftest.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,7 @@
11
import pytest
22

33
from autointent import Context, Dataset
4-
from autointent.configs import (
5-
LoggingConfig,
6-
VectorIndexConfig,
7-
)
4+
from autointent.configs import DataConfig, LoggingConfig, VectorIndexConfig
85
from autointent.nodes import NodeOptimizer
96
from 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

0 commit comments

Comments
 (0)