formatting

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

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...")

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