From 5940c8d908b12d8c99cc03305dec4fcf8bcc3dd8 Mon Sep 17 00:00:00 2001 From: lidanyang Date: Thu, 14 Dec 2023 12:56:01 +0800 Subject: [PATCH] remove old comments --- metagpt/roles/ml_engineer.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/metagpt/roles/ml_engineer.py b/metagpt/roles/ml_engineer.py index 0b76711f4..51faf1e0d 100644 --- a/metagpt/roles/ml_engineer.py +++ b/metagpt/roles/ml_engineer.py @@ -261,14 +261,12 @@ class MLEngineer(Role): """find useful memories only to reduce context length and improve performance""" # TODO dataset description , code steps if task_exclude_field is None: + # Shorten the context as we don't need code steps after we get the codes. + # This doesn't affect current_task below, which should hold the code steps task_exclude_field = {'code_steps'} user_requirement = self.plan.goal data_desc = self.plan.context tasks = [task.dict(exclude=task_exclude_field) for task in self.plan.tasks] - # for task in tasks: - # # Shorten the context as we don't need code steps after we get the codes. - # # This doesn't affect current_task below, which should hold the code steps - # task.pop("code_steps") tasks = json.dumps(tasks, indent=4, ensure_ascii=False) current_task = self.plan.current_task.json() if self.plan.current_task else {} context = STRUCTURAL_CONTEXT.format(