Skip to content

Commit d8d7852

Browse files
committed
enable validation back
1 parent d34a4c0 commit d8d7852

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

autointent/_pipeline/_pipeline.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
from autointent.custom_types import ListOfGenericLabels, NodeType, SamplerType, ValidationScheme
1414
from autointent.metrics import PREDICTION_METRICS_MULTILABEL
1515
from autointent.nodes import InferenceNode, NodeOptimizer
16+
from autointent.nodes.schemes import OptimizationConfig
1617
from autointent.utils import load_default_search_space, load_search_space
1718

1819
from ._schemas import InferencePipelineOutput, InferencePipelineUtteranceOutput
@@ -70,7 +71,7 @@ def from_search_space(cls, search_space: list[dict[str, Any]] | Path | str, seed
7071
"""
7172
if isinstance(search_space, Path | str):
7273
search_space = load_search_space(search_space)
73-
validated_search_space = search_space # OptimizationConfig(search_space).model_dump() # type: ignore[arg-type]
74+
validated_search_space = OptimizationConfig(search_space).model_dump() # type: ignore[arg-type]
7475
nodes = [NodeOptimizer(**node) for node in validated_search_space]
7576
return cls(nodes=nodes, seed=seed)
7677

0 commit comments

Comments
 (0)