Skip to content

Commit 8f30ec9

Browse files
committed
remove regexp validator
1 parent bf9074f commit 8f30ec9

File tree

1 file changed

+2
-17
lines changed

1 file changed

+2
-17
lines changed

autointent/nodes/schemes.py

Lines changed: 2 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
from autointent.custom_types import NodeType
1010
from autointent.modules.abc import Module
11-
from autointent.nodes import DecisionNodeInfo, EmbeddingNodeInfo, RegExpNodeInfo, ScoringNodeInfo
11+
from autointent.nodes import DecisionNodeInfo, EmbeddingNodeInfo, ScoringNodeInfo
1212

1313

1414
def 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

11297
class OptimizationConfig(RootModel[list[SearchSpaceTypes]]):

0 commit comments

Comments
 (0)