From 2ec4f6832422faebf20ca413b1ebf78547a4855d Mon Sep 17 00:00:00 2001
From: Jiayuan Gu <17827258+Jiayuan-Gu@users.noreply.github.com>
Date: Sun, 5 Jun 2022 09:58:33 -0700
Subject: [PATCH] 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
---
 habitat_baselines/utils/common.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/habitat_baselines/utils/common.py b/habitat_baselines/utils/common.py
index 8e780b7386..932c696d5f 100644
--- a/habitat_baselines/utils/common.py
+++ b/habitat_baselines/utils/common.py
@@ -394,7 +394,9 @@ def generate_video(
     )
     if "disk" in video_option:
         assert video_dir is not None
-        images_to_video(images, video_dir, video_name, verbose=verbose)
+        images_to_video(
+            images, video_dir, video_name, fps=fps, verbose=verbose
+        )
     if "tensorboard" in video_option:
         tb_writer.add_video_from_np_images(
             f"episode{episode_id}", checkpoint_idx, images, fps=fps