def _reset(self): motor_state = RoboschoolMujocoXmlEnv._reset(self) aux_state = np.array([motor_state[-3], motor_state[-1]]) return self._render("rgb_array", False), { 'motor': motor_state, "aux": aux_state }
def _reset(self): motor_state = RoboschoolMujocoXmlEnv._reset(self) return self._render("rgb_array", False) # , {'motor': motor_state}