From 60ec58f5df8de1cccc0082145432715a8a94f26a Mon Sep 17 00:00:00 2001
From: you-n-g <you-n-g@users.noreply.github.com>
Date: Fri, 16 May 2025 02:05:15 +0800
Subject: [PATCH] fix: update DS env setup with competition volume and timeout
 (#878)

* refactor: update DS env setup with competition volume and timeout

* refactor: update volume mapping and timeout based on run type
---
 rdagent/components/coder/data_science/share/eval.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/rdagent/components/coder/data_science/share/eval.py b/rdagent/components/coder/data_science/share/eval.py
index d10f44e1..be9b77a3 100644
--- a/rdagent/components/coder/data_science/share/eval.py
+++ b/rdagent/components/coder/data_science/share/eval.py
@@ -43,8 +43,17 @@ class ModelDumpEvaluator(CoSTEEREvaluator):
                 code=err_msg,
                 final_decision=False,
             )
-        env = get_ds_env()
-        env.conf.extra_volumes = {self.scen.debug_path: T("scenarios.data_science.share:scen.input_path").r()}
+        data_source_path = (
+            f"{DS_RD_SETTING.local_data_path}/{self.scen.competition}"
+            if self.data_type == "full"
+            else self.scen.debug_path
+        )
+        env = get_ds_env(
+            extra_volumes={data_source_path: T("scenarios.data_science.share:scen.input_path").r()},
+            running_timeout_period=(
+                DS_RD_SETTING.full_timeout if self.data_type == "full" else DS_RD_SETTING.debug_timeout
+            ),
+        )
 
         # 2) check the result and stdout after reruning the model.