def _render()

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

• 10 lines of code
• 5 McCabe index (conditional complexity)

    def _render(self, mode='human', close=False):
        if close:
            if self.viewer is not None:
                self.viewer = None
            return

        if mode == 'rgb_array':
            self.viewer_setup()
            return _read_pixels(self.sim, *self.buffer_size)
        elif mode == 'human':
            self._get_viewer().render()