diff --git a/rdagent/components/coder/factor_coder/config.py b/rdagent/components/coder/factor_coder/config.py index a09ae666a..809d6a9d5 100644 --- a/rdagent/components/coder/factor_coder/config.py +++ b/rdagent/components/coder/factor_coder/config.py @@ -31,14 +31,15 @@ class FactorCoSTEERSettings(CoSTEERSettings): def get_factor_env( conf_type: Optional[str] = None, - extra_volumes: dict = {}, + extra_volumes: Optional[dict] = None, running_timeout_period: int = 600, enable_cache: Optional[bool] = None, ) -> Env: conf = FactorCoSTEERSettings() if hasattr(conf, "python_bin"): env = LocalEnv(conf=(CondaConf(conda_env_name=os.environ.get("CONDA_DEFAULT_ENV")))) - env.conf.extra_volumes = extra_volumes.copy() + if extra_volumes is not None: + env.conf.extra_volumes = extra_volumes.copy() env.conf.running_timeout_period = running_timeout_period if enable_cache is not None: env.conf.enable_cache = enable_cache diff --git a/rdagent/components/coder/model_coder/conf.py b/rdagent/components/coder/model_coder/conf.py index 1cb8d2d35..e7251aa9d 100644 --- a/rdagent/components/coder/model_coder/conf.py +++ b/rdagent/components/coder/model_coder/conf.py @@ -15,7 +15,7 @@ class ModelCoSTEERSettings(CoSTEERSettings): def get_model_env( conf_type: Optional[str] = None, - extra_volumes: dict = {}, + extra_volumes: Optional[dict] = None, running_timeout_period: int = 600, enable_cache: Optional[bool] = None, ) -> Env: @@ -27,7 +27,8 @@ def get_model_env( else: raise ValueError(f"Unknown env type: {conf.env_type}") - env.conf.extra_volumes = extra_volumes.copy() + if extra_volumes is not None: + env.conf.extra_volumes = extra_volumes.copy() env.conf.running_timeout_period = running_timeout_period if enable_cache is not None: env.conf.enable_cache = enable_cache