Merge branch 'bruce_port_envs' into future

This commit is contained in:
Dominik Moritz Roth 2024-02-02 17:00:35 +01:00
commit 9295e21d68

View File

@ -50,6 +50,7 @@ class BoxPushingEnvBase(MujocoEnv, utils.EzPickle):
self._desired_rod_quat = desired_rod_quat
self._episode_energy = 0.
self.velocity_profile = []
self.observation_space = spaces.Box(
low=-np.inf, high=np.inf, shape=(28,), dtype=np.float64
@ -68,6 +69,8 @@ class BoxPushingEnvBase(MujocoEnv, utils.EzPickle):
unstable_simulation = False
self.velocity_profile.append(self.data.qvel[:7].copy())
try:
self.do_simulation(resultant_action, self.frame_skip)
except Exception as e: