def _get_image()

in gym3/interactive.py [0:0]

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

    def _get_image(self) -> Optional[np.ndarray]:
        """
        Get the image that we should display to the user for the current step
        """
        _, ob, _ = self._env.observe()
        if self._info_key is None:
            if self._ob_key is not None:
                ob = ob[self._ob_key]
            return ob[0]
        else:
            info = self._env.get_info()
            return info[0].get(self._info_key)