From 2faf0f3090e510ea97bae24d8fb3a3fd97039f9e Mon Sep 17 00:00:00 2001 From: erik Date: Tue, 13 Jan 2026 11:43:48 +0200 Subject: [PATCH 1/3] Feat: Add Original Column generator as a default feature generator Added the original column generator to the api defaults file. --- agentune/api/defaults.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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() From 87d48d5d58e64c1bdf641239c6b53a01af2e4480 Mon Sep 17 00:00:00 2001 From: erik Date: Tue, 13 Jan 2026 12:39:08 +0200 Subject: [PATCH 2/3] Remove check relying on the fact features can't be found --- tests/agentune/api/test_ops.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tests/agentune/api/test_ops.py b/tests/agentune/api/test_ops.py index 8f245a82..8c029301 100644 --- a/tests/agentune/api/test_ops.py +++ b/tests/agentune/api/test_ops.py @@ -43,9 +43,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() From 1971820f438a294134c45c9a7d9eef22f7a09827 Mon Sep 17 00:00:00 2001 From: erik Date: Tue, 13 Jan 2026 15:41:33 +0200 Subject: [PATCH 3/3] fix ruff --- tests/agentune/api/test_ops.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/agentune/api/test_ops.py b/tests/agentune/api/test_ops.py index 8c029301..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