gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-17 20:39:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

MujocoEnv: Include shape in error message (#83)

Browse Source
This commit is contained in:
Ilya Kamen
2022-10-27 15:41:42 +02:00
committed by GitHub
parent 5ec39806cf
commit 086964dfa9
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
gymnasium/envs/mujoco/mujoco_env.py
View File

@@ -153,7 +153,9 @@ class BaseMujocoEnv(gym.Env):
""" """
# Check control input is contained in the action space # Check control input is contained in the action space
if np.array(ctrl).shape != self.action_space.shape: if np.array(ctrl).shape != self.action_space.shape:
raise ValueError("Action dimension mismatch") raise ValueError(
f"Action dimension mismatch. Expected {self.action_space.shape}, found {np.array(ctrl).shape}"
)
self._step_mujoco_simulation(ctrl, n_frames) self._step_mujoco_simulation(ctrl, n_frames)
def close(self): def close(self):