From 36bfbf20b62222380263aa77d9dce8571767581e Mon Sep 17 00:00:00 2001 From: Way2Learn <118058822+Xisen-Wang@users.noreply.github.com> Date: Tue, 24 Sep 2024 14:50:01 +0800 Subject: [PATCH] fix: Use fixed file name in model costeer & fixing cache (#311) * use fixed name in model costeer to enable cache * fix ci --- rdagent/components/coder/model_coder/model.py | 1 + rdagent/utils/env.py | 3 ++- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/rdagent/components/coder/model_coder/model.py b/rdagent/components/coder/model_coder/model.py index eca3b432..a8f700e9 100644 --- a/rdagent/components/coder/model_coder/model.py +++ b/rdagent/components/coder/model_coder/model.py @@ -115,6 +115,7 @@ PARAM_INIT_VALUE = {param_init_value} dump_file_names=["execution_feedback_str.pkl", "execution_model_output.pkl"], local_path=str(self.workspace_path), env={}, + code_dump_file_py_name="model_test", ) if results is None: raise RuntimeError(f"Error in running the model code: {log}") diff --git a/rdagent/utils/env.py b/rdagent/utils/env.py index 7c43d743..815dfcde 100644 --- a/rdagent/utils/env.py +++ b/rdagent/utils/env.py @@ -327,11 +327,12 @@ class DockerEnv(Env[DockerConf]): local_path: str | None = None, env: dict | None = None, running_extra_volume: dict | None = None, + code_dump_file_py_name: Optional[str] = None, ): """ Dump the code into the local path and run the code. """ - random_file_name = f"{uuid.uuid4()}.py" + random_file_name = f"{uuid.uuid4()}.py" if code_dump_file_py_name is None else f"{code_dump_file_py_name}.py" with open(os.path.join(local_path, random_file_name), "w") as f: f.write(code) entry = f"python {random_file_name}"