From db34c6f458e977e48872c0dc9f97aa98a1caf4a4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8E=98=E6=9D=83=20=E9=A9=AC?= <mashenquan@fuzhi.ai>
Date: Tue, 30 Apr 2024 12:17:37 +0800
Subject: [PATCH] refactor: PEP8

---
 metagpt/roles/engineer.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/metagpt/roles/engineer.py b/metagpt/roles/engineer.py
index 8959b437a..e7b5cf219 100644
--- a/metagpt/roles/engineer.py
+++ b/metagpt/roles/engineer.py
@@ -332,12 +332,11 @@ class Engineer(Role):
     async def _new_code_actions(self):
         bug_fix = await self._is_fixbug()
         # Prepare file repos
-        changed_src_files = (
-            {self.context.kwargs.src_filename: ChangeType.UNTRACTED}
-            if self.context.kwargs.src_filename
-            else self.project_repo.srcs.changed_files
-        )
-        changed_src_files = self.project_repo.srcs.all_files if bug_fix else changed_src_files
+        changed_src_files = self.project_repo.srcs.changed_files
+        if self.context.kwargs.src_filename:
+            changed_src_files = {self.context.kwargs.src_filename: ChangeType.UNTRACTED}
+        if bug_fix:
+            changed_src_files = self.project_repo.srcs.all_files
         changed_task_files = self.project_repo.docs.task.changed_files
         changed_files = Documents()
         # Recode caused by upstream changes.