style: fix type checker

Author: juelg

python/rcs/envs/sim.py

@@ -199,8 +199,8 @@ def env_from_xml_paths(
 class RandomCubePos(SimWrapper):
     """Wrapper to randomly place cube in the lab environments."""
 
-    def __init__(self, env: gym.Env, include_rotation: bool = False):
-        super().__init__(env)
+    def __init__(self, env: gym.Env, simulation: sim.Sim, include_rotation: bool = False):
+        super().__init__(env, simulation)
         self.include_rotation = include_rotation
 
     def reset(
@@ -268,7 +268,7 @@ def step(self, action: dict[str, Any]):
             # reward += static_reward * info["is_obj_placed"]
 
         # normalize
-        reward /= 5
+        reward /= 5  # type: ignore
         return obs, reward, success, truncated, info