Skip to content

Commit 2ec4f68

Browse files
authored
Fix a minor bug about generate_video (#879)
* Fix a minor bug about `generate_video` `fps` is not passed to `images_to_video` * Update common.py Follow black format
1 parent 5ce36a6 commit 2ec4f68

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

habitat_baselines/utils/common.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -394,7 +394,9 @@ def generate_video(
394394
)
395395
if "disk" in video_option:
396396
assert video_dir is not None
397-
images_to_video(images, video_dir, video_name, verbose=verbose)
397+
images_to_video(
398+
images, video_dir, video_name, fps=fps, verbose=verbose
399+
)
398400
if "tensorboard" in video_option:
399401
tb_writer.add_video_from_np_images(
400402
f"episode{episode_id}", checkpoint_idx, images, fps=fps

0 commit comments

Comments
 (0)