def get_obs()

in gym_xarm/tasks/base.py [0:0]

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

    def get_obs(self):
        if self.obs_type == "state":
            return self._get_obs()
        pixels = self._render(renderer=self.observation_renderer)
        if self.obs_type == "pixels":
            return pixels
        elif self.obs_type == "pixels_agent_pos":
            return {
                "pixels": pixels,
                "agent_pos": self.robot_state,
            }
        else:
            raise ValueError(
                f"Unknown obs_type {self.obs_type}. Must be one of [pixels, state, pixels_agent_pos]"
            )