From 95bed6e8e87fab3b0f600f98a7a236e8076e6451 Mon Sep 17 00:00:00 2001
From: didi <84363704+didiforgithub@users.noreply.github.com>
Date: Fri, 25 Oct 2024 16:56:38 +0800
Subject: [PATCH] modify reproduce function

---
 examples/aflow/experiments/optimize_drop.py      | 12 ++++++++++++
 examples/aflow/experiments/optimize_gsm8k.py     | 12 ++++++++++++
 examples/aflow/experiments/optimize_hotpotqa.py  | 12 ++++++++++++
 examples/aflow/experiments/optimize_humaneval.py | 12 ++++++++++++
 examples/aflow/experiments/optimize_math.py      | 12 ++++++++++++
 examples/aflow/experiments/optimize_mbpp.py      | 12 ++++++++++++
 6 files changed, 72 insertions(+)

diff --git a/examples/aflow/experiments/optimize_drop.py b/examples/aflow/experiments/optimize_drop.py
index 46e53b81a..1b3bcdbcf 100644
--- a/examples/aflow/experiments/optimize_drop.py
+++ b/examples/aflow/experiments/optimize_drop.py
@@ -3,6 +3,18 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
+import os
+import sys
+
+
+def setup_environment():
+    current_path = os.path.abspath(__file__)
+    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
+    sys.path.insert(0, root_path)
+    os.chdir(root_path)
+
+
+setup_environment()
 
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
diff --git a/examples/aflow/experiments/optimize_gsm8k.py b/examples/aflow/experiments/optimize_gsm8k.py
index 008783eac..aefc5ef73 100644
--- a/examples/aflow/experiments/optimize_gsm8k.py
+++ b/examples/aflow/experiments/optimize_gsm8k.py
@@ -3,6 +3,18 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
+import os
+import sys
+
+
+def setup_environment():
+    current_path = os.path.abspath(__file__)
+    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
+    sys.path.insert(0, root_path)
+    os.chdir(root_path)
+
+
+setup_environment()
 
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
diff --git a/examples/aflow/experiments/optimize_hotpotqa.py b/examples/aflow/experiments/optimize_hotpotqa.py
index 7d4a0230a..4a592d116 100644
--- a/examples/aflow/experiments/optimize_hotpotqa.py
+++ b/examples/aflow/experiments/optimize_hotpotqa.py
@@ -3,6 +3,18 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
+import os
+import sys
+
+
+def setup_environment():
+    current_path = os.path.abspath(__file__)
+    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
+    sys.path.insert(0, root_path)
+    os.chdir(root_path)
+
+
+setup_environment()
 
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
diff --git a/examples/aflow/experiments/optimize_humaneval.py b/examples/aflow/experiments/optimize_humaneval.py
index ef99f8210..965f7b355 100644
--- a/examples/aflow/experiments/optimize_humaneval.py
+++ b/examples/aflow/experiments/optimize_humaneval.py
@@ -3,6 +3,18 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
+import os
+import sys
+
+
+def setup_environment():
+    current_path = os.path.abspath(__file__)
+    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
+    sys.path.insert(0, root_path)
+    os.chdir(root_path)
+
+
+setup_environment()
 
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
diff --git a/examples/aflow/experiments/optimize_math.py b/examples/aflow/experiments/optimize_math.py
index d049e2c88..c95e0ad87 100644
--- a/examples/aflow/experiments/optimize_math.py
+++ b/examples/aflow/experiments/optimize_math.py
@@ -3,6 +3,18 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
+import os
+import sys
+
+
+def setup_environment():
+    current_path = os.path.abspath(__file__)
+    root_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.dirname(current_path))))
+    sys.path.insert(0, root_path)
+    os.chdir(root_path)
+
+
+setup_environment()
 
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType
diff --git a/examples/aflow/experiments/optimize_mbpp.py b/examples/aflow/experiments/optimize_mbpp.py
index de71abaec..5f7668762 100644
--- a/examples/aflow/experiments/optimize_mbpp.py
+++ b/examples/aflow/experiments/optimize_mbpp.py
@@ -3,6 +3,18 @@
 # @Author  : didi
 # @Desc    : Entrance of AFlow.
 
+import os
+import sys
+
+
+def setup_environment():
+    current_path = os.path.abspath(__file__)
+    root_path = os.path.dirname(os.path.dirname(os.path.dirname(current_path)))
+    sys.path.insert(0, root_path)
+    os.chdir(root_path)
+
+
+setup_environment()
 
 from metagpt.configs.models_config import ModelsConfig
 from metagpt.ext.aflow.scripts.optimizer import DatasetType, Optimizer, QuestionType