Fix Video Unit Tests

Summary: A reason our OSS repo tests are failing is because on OSS our vidgear dependency was on v0.3.2 and we were on v0.2.4 They changed the `output_filename` parameter to be `output` between these versions, so updating our code Reviewed By: mayaliliya Differential Revision: D50663349
facebookresearch · Oct 26, 2023 · 2d14c58 · 2d14c58
1 parent c7a0eaf
commit 2d14c58
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/augly/video/augmenters/ffmpeg/base_augmenter.py b/augly/video/augmenters/ffmpeg/base_augmenter.py
@@ -46,7 +46,7 @@ def add_augmenter(
             if video_path == output_path:
                 shutil.copyfile(video_path, tmpfile.name)
                 video_path = tmpfile.name
-            writer = WriteGear(output_filename=output_path, logging=True)
+            writer = WriteGear(output=output_path, logging=True)
             writer.execute_ffmpeg_cmd(self.get_command(video_path, output_path))
             writer.close()
 

diff --git a/augly/video/helpers/ffmpeg.py b/augly/video/helpers/ffmpeg.py
@@ -69,7 +69,7 @@ def combine_frames_and_audio_to_file(
 
 
 def execute_vidgear_command(output_path: str, ffmpeg_command: List[str]) -> None:
-    writer = WriteGear(output_filename=output_path, logging=True)
+    writer = WriteGear(output=output_path, logging=True)
     writer.execute_ffmpeg_cmd(ffmpeg_command)
     writer.close()