reformatted

Author: prishajain1

src/maxdiffusion/generate_ltx2.py

@@ -164,7 +164,9 @@ def run(config, pipeline=None, filename_prefix="", commit_hash=None):
     video_path = f"{filename_prefix}ltx2_output_{getattr(config, 'seed', 0)}_{i}.mp4"
     audio_i = audios[i] if audios is not None else None
 
-    export_to_video_with_audio(video=videos[i], fps=fps, audio=audio_i, audio_sample_rate=audio_sample_rate, output_path=video_path)
+    export_to_video_with_audio(
+        video=videos[i], fps=fps, audio=audio_i, audio_sample_rate=audio_sample_rate, output_path=video_path
+    )
 
     saved_video_path.append(video_path)
     if config.output_dir.startswith("gs://"):

src/maxdiffusion/utils/export_utils.py

@@ -232,6 +232,7 @@ def _prepare_audio_stream(container, audio_sample_rate: int):
   Prepare the audio stream for writing.
   """
   from fractions import Fraction
+
   audio_stream = container.add_stream("aac", rate=audio_sample_rate)
   audio_stream.codec_context.sample_rate = audio_sample_rate
   audio_stream.codec_context.layout = "stereo"
@@ -300,7 +301,9 @@ def _write_audio(
   _resample_audio(container, audio_stream, frame_in)
 
 
-def export_to_video_with_audio(video: Any, fps: int, audio: Optional[Any], audio_sample_rate: Optional[int], output_path: str) -> None:
+def export_to_video_with_audio(
+    video: Any, fps: int, audio: Optional[Any], audio_sample_rate: Optional[int], output_path: str
+) -> None:
   """
   Encodes video (and optionally audio) to a file using PyAV.
   Args: