gavindiaz/NexQuant
1
0
Fork 0
mirror of https://github.com/NicolasBohn/NexQuant.git synced 2026-08-06 11:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: fit more competition (#723)

Browse Source 
1. Remove potential <code> tags from the generated code.
2. Use return codes in data_loader, feature, and model.
3. Configure the debug timeout.
This commit is contained in:
Tim
2025-04-10 17:56:57 +08:00
committed by GitHub
parent b5631b9f78
commit f06e6986c8
11 changed files with 53 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rdagent/components/coder/data_science/conf.py
+7 -1
View File
@@ -24,7 +24,11 @@ class DSCoderCoSTEERSettings(CoSTEERSettings):
# TODO: extract a function for env and conf.
def get_ds_env(conf_type: Literal["kaggle", "mlebench"] = "kaggle") -> Env:
def get_ds_env(
conf_type: Literal["kaggle", "mlebench"] = "kaggle",
extra_volumes: dict = {},
running_timeout_period: int = DS_RD_SETTING.debug_timeout,
) -> Env:
"""
Retrieve the appropriate environment configuration based on the env_type setting.
@@ -48,6 +52,8 @@ def get_ds_env(conf_type: Literal["kaggle", "mlebench"] = "kaggle") -> Env:
)
else:
raise ValueError(f"Unknown env type: {conf.env_type}")
env.conf.extra_volumes = extra_volumes
env.conf.running_timeout_period = running_timeout_period
return env