Skip to content

Commit 4172a96

Browse files
committed
fix: satisfy mypy for playback snapshot contract
1 parent 1cbaf49 commit 4172a96

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

‎src/unilab/base/np_env.py‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -228,7 +228,11 @@ def get_physics_state_snapshot(self) -> np.ndarray:
228228
raise NotImplementedError(
229229
f"{self._backend.__class__.__name__} does not support physics-state playback"
230230
)
231-
return np.asarray(self._backend.get_physics_state(), dtype=np.float32).copy()
231+
physics_state = cast(
232+
np.ndarray, np.asarray(self._backend.get_physics_state(), dtype=np.float32)
233+
)
234+
snapshot = cast(np.ndarray, physics_state.copy())
235+
return snapshot
232236

233237
@abc.abstractmethod
234238
def apply_action(self, actions: np.ndarray, state: NpEnvState) -> np.ndarray:

0 commit comments

Comments
 (0)