def _get_image()

in gym3/viewer.py [0:0]

• 9 lines of code
• 3 McCabe index (conditional complexity)

    def _get_image(self) -> None:
        _, ob, _ = self.observe()
        if self._info_key is None:
            if self._ob_key is not None:
                ob = ob[self._ob_key]
            return ob[self._env_index]
        else:
            info = self.get_info()
            return info[self._env_index][self._info_key]