def _append_observation()

in gym3/video_recorder.py [0:0]

• 19 lines of code
• 7 McCabe index (conditional complexity)

    def _append_observation(self) -> None:
        _, ob, _ = self.observe()
        if self._info_key is None:
            if self._ob_key is not None:
                ob = ob[self._ob_key]
            img = ob[self._env_index]
        else:
            info = self.get_info()
            img = info[self._env_index].get(self._info_key)
            # the first info for a converted environment may be empty
            if self._first_step and img is None:
                return
        frame = self._process_frame(img.astype(np.uint8))
        self._writer.append_data(frame)
        if self._renderer is not None:
            self._renderer.start()
            self._renderer.draw_bitmap(
                0, 0, self._renderer.width, self._renderer.height, image=frame
            )
            self._renderer.finish()