diff --git a/agentune/api/defaults.py b/agentune/api/defaults.py index 91aea526..32697bb5 100644 --- a/agentune/api/defaults.py +++ b/agentune/api/defaults.py @@ -19,6 +19,9 @@ from agentune.analyze.feature.gen.insightful_text_generator.insightful_text_generator import ( ConversationQueryFeatureGenerator, ) +from agentune.analyze.feature.gen.original_column_generator.original_column_generator import ( + OriginalColumnsGenerator, +) from agentune.analyze.feature.select import FeatureSelector from agentune.analyze.feature.select.base import EnrichedFeatureSelector from agentune.analyze.feature.select.linear_pairwise import LinearPairWiseFeatureSelector @@ -90,7 +93,7 @@ def conversation_query_feature_generator(self) -> ConversationQueryFeatureGenera return ConversationQueryFeatureGenerator.default(self.run_context._llm_context) def feature_generators(self) -> tuple[FeatureGenerator, ...]: - return (self.conversation_query_feature_generator(), ) + return (self.conversation_query_feature_generator(), OriginalColumnsGenerator()) def feature_selector(self) -> FeatureSelector | EnrichedFeatureSelector: return LinearPairWiseFeatureSelector() diff --git a/tests/agentune/api/test_ops.py b/tests/agentune/api/test_ops.py index 8f245a82..00cdf81e 100644 --- a/tests/agentune/api/test_ops.py +++ b/tests/agentune/api/test_ops.py @@ -12,7 +12,6 @@ from agentune.analyze.feature.problem import ProblemDescription from agentune.analyze.run.analysis.base import ( AnalyzeComponents, - NoFeaturesFoundError, ) from agentune.api.base import RunContext from agentune.api.data import BoundTable @@ -43,9 +42,6 @@ async def test_e2e_flow_synthetic(input_data_csv_path: Path, tmp_path: Path) -> split_input = await input.split() problem_description = ProblemDescription('target', 'Test synthetic data problem') - with pytest.raises(NoFeaturesFoundError): # Default generators can't do anything with this synthetic data - await ctx.ops.analyze(problem_description, split_input) - components = AnalyzeComponents( generators=(ToySyncFeatureGenerator(), ToyAsyncFeatureGenerator(), ToySyncFeatureGenerator(), ToyAsyncFeatureGenerator()), selector=ToyAsyncEnrichedFeatureSelector()