From c47de22c3deac1a979abb2faed48e761a526982e Mon Sep 17 00:00:00 2001 From: XianBW <36835909+XianBW@users.noreply.github.com> Date: Thu, 20 Mar 2025 19:15:37 +0800 Subject: [PATCH] add task type logger tag in coding (#714) --- rdagent/app/data_science/loop.py | 25 +++++++++++++------------ 1 file changed, 13 insertions(+), 12 deletions(-) diff --git a/rdagent/app/data_science/loop.py b/rdagent/app/data_science/loop.py index 2c974025..b1c5873c 100644 --- a/rdagent/app/data_science/loop.py +++ b/rdagent/app/data_science/loop.py @@ -75,18 +75,19 @@ class DataScienceRDLoop(RDLoop): exp = prev_out["direct_exp_gen"] for tasks in exp.pending_tasks_list: exp.sub_tasks = tasks - if isinstance(exp.sub_tasks[0], DataLoaderTask): - exp = self.data_loader_coder.develop(exp) - elif isinstance(exp.sub_tasks[0], FeatureTask): - exp = self.feature_coder.develop(exp) - elif isinstance(exp.sub_tasks[0], ModelTask): - exp = self.model_coder.develop(exp) - elif isinstance(exp.sub_tasks[0], EnsembleTask): - exp = self.ensemble_coder.develop(exp) - elif isinstance(exp.sub_tasks[0], WorkflowTask): - exp = self.workflow_coder.develop(exp) - else: - raise NotImplementedError(f"Unsupported component in DataScienceRDLoop: {exp.hypothesis.component}") + with logger.tag(f"{exp.sub_tasks[0].__class__.__name__}"): + if isinstance(exp.sub_tasks[0], DataLoaderTask): + exp = self.data_loader_coder.develop(exp) + elif isinstance(exp.sub_tasks[0], FeatureTask): + exp = self.feature_coder.develop(exp) + elif isinstance(exp.sub_tasks[0], ModelTask): + exp = self.model_coder.develop(exp) + elif isinstance(exp.sub_tasks[0], EnsembleTask): + exp = self.ensemble_coder.develop(exp) + elif isinstance(exp.sub_tasks[0], WorkflowTask): + exp = self.workflow_coder.develop(exp) + else: + raise NotImplementedError(f"Unsupported component in DataScienceRDLoop: {exp.hypothesis.component}") exp.sub_tasks = [] logger.log_object(exp) return exp