def _draw()

in gym3/interactive.py [0:0]

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

    def _draw(self) -> None:
        if self._display_info_seconds_remaining > 0:
            text = "=== episode complete ===\n\n" + self._format_info()
            self._renderer.draw_text(
                self._renderer.width // 2,
                self._renderer.height // 2,
                text=text,
                centered=True,
            )
        else:
            image = self._get_image()
            if image is None:
                self._renderer.draw_text(
                    self._renderer.width // 2,
                    self._renderer.height // 2,
                    text="(missing image)",
                    centered=True,
                )
            else:
                assert (
                    len(image.shape) == 3 and image.shape[-1] == 3
                ), "expected (H, W, C) RGB image with C = 3"
                assert image.dtype.name == "uint8", "expected uint8 image"
                self._draw_step()