From 802ba8f4b4aae01b2d2b91598cda229a3c2fd396 Mon Sep 17 00:00:00 2001 From: yuecideng Date: Wed, 25 Mar 2026 06:11:12 +0000 Subject: [PATCH] wip --- embodichain/lab/gym/envs/managers/record.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/embodichain/lab/gym/envs/managers/record.py b/embodichain/lab/gym/envs/managers/record.py index 588d14d8..4404035f 100644 --- a/embodichain/lab/gym/envs/managers/record.py +++ b/embodichain/lab/gym/envs/managers/record.py @@ -144,7 +144,7 @@ def __call__( ): # TODO: the current implementation will lost the final episode frames recording. # Check if the frames should be saved for the current episode - if env.elapsed_steps.sum().item() == len(env_ids) and len(self._frames) > 0: + if env.elapsed_steps.sum().item() == 0 and len(self._frames) > 0: video_name = f"episode_{self._current_episode}_{self._name}" images_to_video(self._frames, save_path, video_name, fps=20)