File tree Expand file tree Collapse file tree 1 file changed +2
-17
lines changed
Expand file tree Collapse file tree 1 file changed +2
-17
lines changed Original file line number Diff line number Diff line change 88
99from autointent .custom_types import NodeType
1010from autointent .modules .abc import Module
11- from autointent .nodes import DecisionNodeInfo , EmbeddingNodeInfo , RegExpNodeInfo , ScoringNodeInfo
11+ from autointent .nodes import DecisionNodeInfo , EmbeddingNodeInfo , ScoringNodeInfo
1212
1313
1414def generate_models_and_union_type_for_classes (
@@ -91,22 +91,7 @@ class ScoringNodeValidator(BaseModel):
9191 search_space : list [ScoringSearchSpaceType ]
9292
9393
94- RegexpSearchSpaceType : TypeAlias = generate_models_and_union_type_for_classes ( # type: ignore[valid-type]
95- list (RegExpNodeInfo .modules_available .values ())
96- )
97- RegexpMetrics : TypeAlias = Literal [tuple (RegExpNodeInfo .metrics_available .keys ())] # type: ignore[valid-type]
98-
99-
100- class RegexNodeValidator (BaseModel ):
101- """Search space configuration for the Regexp node."""
102-
103- node_type : NodeType = NodeType .regexp
104- target_metric : RegexpMetrics
105- metrics : list [RegexpMetrics ] | None = None
106- search_space : list [RegexpSearchSpaceType ]
107-
108-
109- SearchSpaceTypes : TypeAlias = RegexNodeValidator | EmbeddingNodeValidator | ScoringNodeValidator | DecisionNodeValidator
94+ SearchSpaceTypes : TypeAlias = EmbeddingNodeValidator | ScoringNodeValidator | DecisionNodeValidator
11095
11196
11297class OptimizationConfig (RootModel [list [SearchSpaceTypes ]]):
You can’t perform that action at this time.
0 commit comments