diff --git a/rdagent/components/coder/data_science/pipeline/__init__.py b/rdagent/components/coder/data_science/pipeline/__init__.py index 4dc791e2..64a1cab2 100644 --- a/rdagent/components/coder/data_science/pipeline/__init__.py +++ b/rdagent/components/coder/data_science/pipeline/__init__.py @@ -59,7 +59,6 @@ class PipelineMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy): prev_task_feedback: CoSTEERSingleFeedback | None = None, ) -> dict[str, str]: competition_info = self.scen.get_scenario_all_desc(eda_output=workspace.file_dict.get("EDA.md", None)) - runtime_environment = self.scen.get_runtime_environment() data_folder_info = self.scen.processed_data_folder_description pipeline_task_info = target_task.get_task_information() @@ -85,7 +84,7 @@ class PipelineMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy): queried_similar_successful_knowledge=queried_similar_successful_knowledge, queried_former_failed_knowledge=queried_former_failed_knowledge[0], out_spec=PythonAgentOut.get_spec(), - runtime_environment=runtime_environment, + runtime_environment=self.scen.get_runtime_environment(), package_info=target_task.package_info, enable_model_dump=DS_RD_SETTING.enable_model_dump, enable_debug_mode=DS_RD_SETTING.sample_data_by_LLM, diff --git a/rdagent/components/coder/data_science/raw_data_loader/__init__.py b/rdagent/components/coder/data_science/raw_data_loader/__init__.py index 139d8e82..c7e6046e 100644 --- a/rdagent/components/coder/data_science/raw_data_loader/__init__.py +++ b/rdagent/components/coder/data_science/raw_data_loader/__init__.py @@ -68,7 +68,6 @@ class DataLoaderMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy): # return a workspace with "load_data.py", "spec/load_data.md" inside # assign the implemented code to the new workspace. competition_info = self.scen.get_scenario_all_desc(eda_output=workspace.file_dict.get("EDA.md", None)) - runtime_environment = self.scen.get_runtime_environment() data_folder_info = self.scen.processed_data_folder_description data_loader_task_info = target_task.get_task_information() @@ -96,7 +95,7 @@ class DataLoaderMultiProcessEvolvingStrategy(MultiProcessEvolvingStrategy): if DS_RD_SETTING.spec_enabled: if "spec/data_loader.md" not in workspace.file_dict: # Only generate the spec once system_prompt = T(".prompts:spec.system").r( - runtime_environment=runtime_environment, + runtime_environment=self.scen.get_runtime_environment(), task_desc=data_loader_task_info, competition_info=competition_info, folder_spec=data_folder_info, diff --git a/rdagent/scenarios/data_science/dev/runner/eval.py b/rdagent/scenarios/data_science/dev/runner/eval.py index 4e3f3d1d..3ba63199 100644 --- a/rdagent/scenarios/data_science/dev/runner/eval.py +++ b/rdagent/scenarios/data_science/dev/runner/eval.py @@ -133,7 +133,7 @@ class DSCoSTEERCoSTEEREvaluator(CoSTEEREvaluator): scenario=self.scen.get_scenario_all_desc(eda_output=implementation.file_dict.get("EDA.md", None)), is_sub_enabled=test_eval.is_sub_enabled(self.scen.competition), task_desc=target_task.get_task_information(), - runtime_environment=self.scen.runtime_environment, + runtime_environment=self.scen.get_runtime_environment(), ) user_prompt = T(".prompts:DSCoSTEER_eval.user").r( code=implementation.all_codes, diff --git a/rdagent/scenarios/data_science/proposal/exp_gen/draft/draft.py b/rdagent/scenarios/data_science/proposal/exp_gen/draft/draft.py index ff42a022..cf6a6bb8 100644 --- a/rdagent/scenarios/data_science/proposal/exp_gen/draft/draft.py +++ b/rdagent/scenarios/data_science/proposal/exp_gen/draft/draft.py @@ -144,9 +144,8 @@ class DSDraftV2ExpGen(ExpGen): return json.loads(response)["tag"].lower() def knowledge_gen(self) -> str: - runtime_environment = self.scen.get_runtime_environment() general_knowledge = T(".prompts_draft:knowledge.general").r( - runtime_environment=runtime_environment, + runtime_environment=self.scen.get_runtime_environment(), component_desc=T(".prompts_draft:description.component_description").r(), ) return f"{general_knowledge}" diff --git a/rdagent/scenarios/data_science/scen/runtime_info.py b/rdagent/scenarios/data_science/scen/runtime_info.py index 144f289b..c1bb0408 100644 --- a/rdagent/scenarios/data_science/scen/runtime_info.py +++ b/rdagent/scenarios/data_science/scen/runtime_info.py @@ -5,6 +5,7 @@ from importlib.metadata import distributions def print_runtime_info(): + print("=== Python Runtime Info ===") print(f"Python {sys.version} on {platform.system()} {platform.release()}")