Ignore mypy errors

Author: araffin

tests/wrappers/test_action_masker.py

@@ -6,7 +6,7 @@
 
 
 class IdentityEnvDiscrete(IdentityEnv):
-    def __init__(self, dim: int = 1, ep_length: int = 100):
+    def __init__(self, dim=1, ep_length=100):
         """
         Identity environment for testing purposes
 
@@ -17,12 +17,12 @@ def __init__(self, dim: int = 1, ep_length: int = 100):
         self.useless_property = 1
         super().__init__(ep_length=ep_length, space=space)
 
-    def _action_masks(self) -> list[int]:
+    def _action_masks(self):  #  -> list[bool]
         assert isinstance(self.action_space, spaces.Discrete)
         return [i == self.state for i in range(self.action_space.n)]
 
 
-def action_mask_fn(env: IdentityEnvDiscrete) -> list[int]:
+def action_mask_fn(env: IdentityEnvDiscrete):  # -> list[int]
     assert isinstance(env.action_space, spaces.Discrete)
     return [i == env.state for i in range(env.action_space.n)]