get ray to stop crashing

This commit is contained in:
Marek Wolan
2024-05-31 15:25:08 +01:00
parent 46ef3b8041
commit e48b71ea1a
5 changed files with 13 additions and 10 deletions

View File

@@ -45,7 +45,8 @@ class PrimaiteRayMARLEnv(MultiAgentEnv):
self.action_space = gymnasium.spaces.Dict(
{name: agent.action_manager.space for name, agent in self.agents.items()}
)
self._obs_space_in_preferred_format = True
self._action_space_in_preferred_format = True
super().__init__()
@property