Save the video from the RLlib example locally

[KennyOrellana]

Hi there! I'm currently trying to save a video from the RLlib example to my local storage, but I'm encountering an error. Would you have any idea what might be going wrong? Any help would be greatly appreciated!

    def on_episode_end(
            self,
            *,
            worker: RolloutWorker,
            base_env: BaseEnv,
            policies: Dict[PolicyID, Policy],
            episode: Episode,
            **kwargs,
    ) -> None:
        # vid = np.transpose(self.frames, (0, 3, 1, 2))
        # episode.media["rendering"] = wandb.Video(
        #     vid, fps=1 / base_env.vector_env.env.world.dt, format="mp4"
        # )

        save_video(self.name, self.frames, 1 / base_env.vector_env.env.world.dt)

        self.frames = []

[matteobettini]

hello,

what is the problem you are encountering?
can you explain the error and link it?