gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-07-27 23:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: minor fix to runtime_environment (#1089)

Browse Source 
* minor fix to funtime_environment

* add python runtime info header
This commit is contained in:
Xu Yang
2025-07-17 19:58:06 +08:00
committed by GitHub
parent 221fea1e15
commit 51921c2344
5 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/pipeline/__init__.py
+1 -2
View File
@@ -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,
rdagent/components/coder/data_science/raw_data_loader/__init__.py
+1 -2
View File
@@ -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,
rdagent/scenarios/data_science/dev/runner/eval.py
+1 -1
View File
@@ -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,
rdagent/scenarios/data_science/proposal/exp_gen/draft/draft.py
+1 -2
View File
@@ -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}"
rdagent/scenarios/data_science/scen/runtime_info.py
+1
View File
@@ -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()}")