From 2d14c58305a988c32853480fee3effa881f3d6bf Mon Sep 17 00:00:00 2001 From: Joanna Bitton Date: Thu, 26 Oct 2023 10:26:55 -0700 Subject: [PATCH] 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 --- augly/video/augmenters/ffmpeg/base_augmenter.py | 2 +- augly/video/helpers/ffmpeg.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/augly/video/augmenters/ffmpeg/base_augmenter.py b/augly/video/augmenters/ffmpeg/base_augmenter.py index 90893c9e..ba48d7bf 100644 --- 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 index b67537fc..bf7e5f34 100644 --- 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()