def render()

in mujoco_worldgen/env.py [0:0]

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

    def render(self, mode='human', close=False):
        if close:
            # TODO: actually close the inspection viewer
            return
        assert self.sim is not None, \
            "Please reset environment before render()."
        if mode == 'human':
            # Use a nicely-interactive version of the mujoco viewer
            if self.viewer is None:
                # Inline import since this is only relevant on platforms
                # which have GLFW support.
                from mujoco_py.mjviewer import MjViewer  # noqa
                self.viewer = MjViewer(self.sim)
            self.viewer.render()
        elif mode == 'rgb_array':
            return self.sim.render(500, 500)
        else:
            raise ValueError("Unsupported mode %s" % mode)