Spaces:

Pulastya0
/

Data-Science-Agent

Running

Pulastya B commited on Jan 28

Commit

ae7f05f

1 Parent(s): e5a2aca

Fixed Agent Reasoning-*1

Files changed (1) hide show

src/orchestrator.py CHANGED Viewed

@@ -3307,7 +3307,7 @@ You receive quality reports from EDA agent and deliver clean data to modeling ag
                                         workflow_state={
                                             'iteration': iteration,
                                             'workflow_history': workflow_history,
-                                            'current_file': self.dataset_path,
                                             'task_description': task_description,
                                             'target_col': target_col
                                         },
@@ -4019,7 +4019,7 @@ You receive quality reports from EDA agent and deliver clean data to modeling ag
                             workflow_state={
                                 'iteration': iteration,
                                 'workflow_history': workflow_history,
-                                'current_file': self.dataset_path,
                                 'task_description': task_description,
                                 'target_col': target_col
                             },

                                         workflow_state={
                                             'iteration': iteration,
                                             'workflow_history': workflow_history,
+                                            'current_file': file_path,
                                             'task_description': task_description,
                                             'target_col': target_col
                                         },
                             workflow_state={
                                 'iteration': iteration,
                                 'workflow_history': workflow_history,
+                                'current_file': file_path,
                                 'task_description': task_description,
                                 'target_col': target_col
                             },