Skip to content

Commit

Permalink
Update walk_v0.py
Browse files Browse the repository at this point in the history
updates to correctly call BaseV0 methods
  • Loading branch information
cameronberg authored Jul 24, 2023
1 parent d29dd52 commit bb2924c
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions myosuite/envs/myo/walk_v0.py
Original file line number Diff line number Diff line change
Expand Up @@ -404,7 +404,7 @@ def __init__(self, model_path, obsd_model_path=None, seed=None, **kwargs):
# first construct the inheritance chain, which is just __init__ calls all the way down, with env_base
# creating the sim / sim_obsd instances. Next we run through "setup" which relies on sim / sim_obsd
# created in __init__ to complete the setup.
super().__init__(model_path=model_path, obsd_model_path=obsd_model_path, seed=seed)
BaseV0.__init__(model_path=model_path, obsd_model_path=obsd_model_path, seed=seed)
self._setup(**kwargs)

def _setup(self,
Expand Down Expand Up @@ -432,7 +432,7 @@ def _setup(self,
self.variant = variant
self.steps = 0

super()._setup(obs_keys=obs_keys,
BaseV0._setup(obs_keys=obs_keys,
weighted_reward_keys=weighted_reward_keys,
**kwargs
)
Expand Down Expand Up @@ -481,7 +481,7 @@ def reset(self):
else:
qpos, qvel = self.sim.model.key_qpos[0], self.sim.model.key_qvel[0]
self.robot.sync_sims(self.sim, self.sim_obsd)
obs = super().reset(reset_qpos=qpos, reset_qvel=qvel)
obs = BaseV0.reset(reset_qpos=qpos, reset_qvel=qvel)
return obs

def _get_done(self):
Expand Down

0 comments on commit bb2924c

Please sign in to comment.