From 4ecd427bea9c42af1595af9fdee4785a7d0a6934 Mon Sep 17 00:00:00 2001
From: yzlin <yzlin@fuzhi.ai>
Date: Thu, 11 Jan 2024 00:47:28 +0800
Subject: [PATCH] formatting

---
 metagpt/roles/code_interpreter.py |  5 +----
 metagpt/roles/ml_engineer.py      | 24 ++----------------------
 metagpt/roles/tool_maker.py       |  4 +---
 3 files changed, 4 insertions(+), 29 deletions(-)

diff --git a/metagpt/roles/code_interpreter.py b/metagpt/roles/code_interpreter.py
index 9bb543d99..6bbd923e6 100644
--- a/metagpt/roles/code_interpreter.py
+++ b/metagpt/roles/code_interpreter.py
@@ -4,10 +4,7 @@ from pydantic import Field
 
 from metagpt.actions.ask_review import ReviewConst
 from metagpt.actions.execute_code import ExecutePyCode
-from metagpt.actions.write_analysis_code import (
-    WriteCodeByGenerate,
-    WriteCodeWithTools,
-)
+from metagpt.actions.write_analysis_code import WriteCodeByGenerate, WriteCodeWithTools
 from metagpt.logs import logger
 from metagpt.roles import Role
 from metagpt.roles.tool_maker import ToolMaker
diff --git a/metagpt/roles/ml_engineer.py b/metagpt/roles/ml_engineer.py
index b6d660137..639a517d6 100644
--- a/metagpt/roles/ml_engineer.py
+++ b/metagpt/roles/ml_engineer.py
@@ -19,27 +19,8 @@ class MLEngineer(CodeInterpreter):
     debug_context: list = []
     latest_code: str = ""
 
-    def __init__(
-        self,
-        name="Mark",
-        profile="MLEngineer",
-        goal="",
-        auto_run=False,
-        use_tools=False,
-        use_code_steps=False,
-        make_udfs=False,
-        use_udfs=False,
-    ):
-        super().__init__(
-            name=name,
-            profile=profile,
-            goal=goal,
-            auto_run=auto_run,
-            use_tools=use_tools,
-            use_code_steps=use_code_steps,
-            make_udfs=make_udfs,
-            use_udfs=use_udfs,
-        )
+    def __init__(self, name="Mark", profile="MLEngineer", **kwargs):
+        super().__init__(name=name, profile=profile, **kwargs)
         # self._watch([DownloadData, SubmitResult])  # in multi-agent settings
 
     async def _plan_and_act(self):
@@ -76,7 +57,6 @@ class MLEngineer(CodeInterpreter):
             return await super()._write_code()
 
         code_execution_count = sum([msg.cause_by == any_to_str(ExecutePyCode) for msg in self.working_memory.get()])
-        print("*" * 10, code_execution_count)
 
         if code_execution_count > 0:
             logger.warning("We got a bug code, now start to debug...")
diff --git a/metagpt/roles/tool_maker.py b/metagpt/roles/tool_maker.py
index a2f854adb..7fec7b739 100644
--- a/metagpt/roles/tool_maker.py
+++ b/metagpt/roles/tool_maker.py
@@ -1,9 +1,7 @@
 from pydantic import Field
 
 from metagpt.actions.execute_code import ExecutePyCode
-from metagpt.actions.write_analysis_code import (
-    MakeTools,
-)
+from metagpt.actions.write_analysis_code import MakeTools
 from metagpt.logs import logger
 from metagpt.roles import Role
 from metagpt.utils.common import remove_comments