def set_euler()

in robosumo/envs/agents.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

    def set_euler(self, euler):
        """Set euler angles the agent; any element can be None."""
        qpos = self._env.data.qpos.ravel().copy()
        start = self.qpos_start_idx
        if euler[0]: qpos[start + 4] = euler[0]
        if euler[1]: qpos[start + 5] = euler[1]
        if euler[2]: qpos[start + 6] = euler[2]
        qvel = self._env.data.qvel.ravel()
        self._env.set_state(qpos, qvel)