Env creation tutorial 3: Change action mask type to int8 (#1038)

Farama-Foundation · Jul 24, 2023 · e9127d3 · e9127d3
1 parent 8e16523
commit e9127d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/EnvironmentCreation/tutorial3_action_masking.py b/tutorials/EnvironmentCreation/tutorial3_action_masking.py
@@ -71,7 +71,7 @@ def step(self, actions):
             self.guard_y += 1
 
         # Generate action masks
-        prisoner_action_mask = np.ones(4)
+        prisoner_action_mask = np.ones(4, dtype=np.int8)
         if self.prisoner_x == 0:
             prisoner_action_mask[0] = 0  # Block left movement
         elif self.prisoner_x == 6:
@@ -81,7 +81,7 @@ def step(self, actions):
         elif self.prisoner_y == 6:
             prisoner_action_mask[3] = 0  # Block up movement
 
-        guard_action_mask = np.ones(4)
+        guard_action_mask = np.ones(4, dtype=np.int8)
         if self.guard_x == 0:
             guard_action_mask[0] = 0
         elif self.guard_x == 6: