✨ update registry
This commit is contained in:
@@ -20,4 +20,5 @@ def build_env(env_name: str, cfg: DictConfig) -> BaseEnv:
|
|||||||
|
|
||||||
env_cls, config_cls = ENV_REGISTRY[env_name]
|
env_cls, config_cls = ENV_REGISTRY[env_name]
|
||||||
env_dict = OmegaConf.to_container(cfg.env, resolve=True)
|
env_dict = OmegaConf.to_container(cfg.env, resolve=True)
|
||||||
|
env_dict.pop("hpo", None) # HPO range metadata — not an env config field
|
||||||
return env_cls(config_cls(**env_dict))
|
return env_cls(config_cls(**env_dict))
|
||||||
|
|||||||
Reference in New Issue
Block a user