Skip to content

Commit bfdc6ce

Browse files
committed
fix: Pass LawAndOrderValidator to BasicFeatureGenerator
After changing BasicFeatureGenerator's validators field to default to an empty tuple (from LawAndOrderValidator), SemanticInsightsGenerator needs to explicitly pass the validator. The docstring already stated it uses LawAndOrderValidator, now the implementation matches.
1 parent def6769 commit bfdc6ce

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

  • agentune/analyze/feature/gen/semantic_insights_generator

agentune/analyze/feature/gen/semantic_insights_generator/generator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@
1212
BasicFeatureGenerator,
1313
)
1414
from agentune.analyze.feature.problem import Problem
15+
from agentune.analyze.feature.validate.law_and_order import LawAndOrderValidator
1516
from agentune.analyze.join.base import TablesWithJoinStrategies
1617
from agentune.core.dataset import Dataset
1718
from agentune.core.sercontext import LLMWithSpec
@@ -59,6 +60,7 @@ async def agenerate(
5960
generation_model=self.generation_model,
6061
repair_model=self.repair_model,
6162
seed=self.seed,
63+
validators=(LawAndOrderValidator(),),
6264
)
6365

6466
async for gen_feature in basic_gen.agenerate(

0 commit comments

Comments
 (0)