2623 Add e2e tests for action masking

2024-07-10 11:01:42 +01:00
parent faf268a9b9
commit 7201b7b8e0
8 changed files with 897 additions and 319 deletions
--- a/src/primaite/session/ray_envs.py
+++ b/src/primaite/session/ray_envs.py
@@ -187,7 +187,7 @@ class PrimaiteRayEnv(gymnasium.Env):
        # if action masking is enabled, intercept the step method and add action mask to observation
        if self.env.agent.action_masking:
            obs, *_ = self.env.step(action)
-            new_obs = {"action_mask": self.env.action_masks(), "observations": obs}
+            new_obs = {"action_mask": self.game.action_mask(self.env._agent_name), "observations": obs}
            return new_obs, *_
        else:
            return self.env.step(action)