NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add back env_config_cls

Browse source
This commit is contained in:
hjc-puro 2025-05-02 09:00:57 -07:00
parent 6661e286c4
commit 7c6c5edf30
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
atroposlib/envs/base.py
View file

@ -155,6 +155,7 @@ class BaseEnvConfig(BaseModel):
class BaseEnv(ABC):
name = None
env_config_cls = BaseEnvConfig
def __init__(
self,
@ -228,7 +229,7 @@ class BaseEnv(ABC):
"""
Initialize the config
"""
return BaseEnvConfig(), ServerBaseline()
return cls.env_config_cls(), ServerBaseline()
async def collect_trajectory(self, item: Item) -> Tuple[Any | None, List[Item]]:
raise NotImplementedError(