Skip to content

Commit 816ada0

Browse files
authored
fix: update DS env setup with competition volume and timeout (microsoft#878)
* refactor: update DS env setup with competition volume and timeout * refactor: update volume mapping and timeout based on run type
1 parent 516cb69 commit 816ada0

File tree

1 file changed

+11
-2
lines changed
  • rdagent/components/coder/data_science/share

1 file changed

+11
-2
lines changed

rdagent/components/coder/data_science/share/eval.py

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,8 +43,17 @@ def evaluate(
4343
code=err_msg,
4444
final_decision=False,
4545
)
46-
env = get_ds_env()
47-
env.conf.extra_volumes = {self.scen.debug_path: T("scenarios.data_science.share:scen.input_path").r()}
46+
data_source_path = (
47+
f"{DS_RD_SETTING.local_data_path}/{self.scen.competition}"
48+
if self.data_type == "full"
49+
else self.scen.debug_path
50+
)
51+
env = get_ds_env(
52+
extra_volumes={data_source_path: T("scenarios.data_science.share:scen.input_path").r()},
53+
running_timeout_period=(
54+
DS_RD_SETTING.full_timeout if self.data_type == "full" else DS_RD_SETTING.debug_timeout
55+
),
56+
)
4857

4958
# 2) check the result and stdout after reruning the model.
5059

0 commit comments

Comments
 (0)