def _get_obs(self):
    obs = super()._get_obs()
    obs['observation'] = np.concatenate([obs['observation'], obs['desired_goal']])
    return obs
def _get_obs(self):
    obs = super()._get_obs()
    obs['observation'] = np.concatenate([obs['observation'], obs['desired_goal']])
    return obs