diff --git a/metagpt/actions/write_code_steps.py b/metagpt/actions/write_code_steps.py
index 47ea0b1df..d3f6e5553 100644
--- a/metagpt/actions/write_code_steps.py
+++ b/metagpt/actions/write_code_steps.py
@@ -72,6 +72,6 @@ class WriteCodeSteps(Action):
         context = STRUCTURAL_CONTEXT.format(
             user_requirement=user_requirement, tasks=tasks, current_task=current_task
         )
-        print(context)
+        # print(context)
         return context
 
diff --git a/metagpt/llm.py b/metagpt/llm.py
index c8ddf9a26..4edcd7a83 100644
--- a/metagpt/llm.py
+++ b/metagpt/llm.py
@@ -11,7 +11,7 @@ from metagpt.config import CONFIG
 from metagpt.provider.anthropic_api import Claude2 as Claude
 from metagpt.provider.openai_api import OpenAIGPTAPI
 from metagpt.provider.zhipuai_api import ZhiPuAIGPTAPI
-# from metagpt.provider.spark_api import SparkAPI
+from metagpt.provider.spark_api import SparkAPI
 from metagpt.provider.human_provider import HumanProvider
 
 
diff --git a/metagpt/roles/ml_engineer.py b/metagpt/roles/ml_engineer.py
index e957d66c4..ce0689497 100644
--- a/metagpt/roles/ml_engineer.py
+++ b/metagpt/roles/ml_engineer.py
@@ -123,10 +123,10 @@ class MLEngineer(Role):
             # breakpoint()
 
             if not self.use_tools or self.plan.current_task.task_type == "other":
-                code = "print('abc')"
-                # code = await WriteCodeByGenerate().run(
-                #     context=context, plan=self.plan, code_steps=code_steps, temperature=0.0
-                # )
+                # code = "print('abc')"
+                code = await WriteCodeByGenerate().run(
+                    context=context, plan=self.plan, code_steps=code_steps, temperature=0.0
+                )
                 cause_by = WriteCodeByGenerate
             else:
                 code = await WriteCodeWithTools().run(