format

mlx_video/models/wan2/postprocess.py

@@ -1,6 +1,8 @@
-import numpy as np
 from pathlib import Path
 
+import numpy as np
+
+
 def save_video(frames: np.ndarray, output_path: str, fps: int = 16):
     """Save video frames to MP4.
 
@@ -11,6 +13,7 @@ def save_video(frames: np.ndarray, output_path: str, fps: int = 16):
     """
     try:
         import imageio
+
         writer = imageio.get_writer(output_path, fps=fps, codec="libx264", quality=8)
         for frame in frames:
             writer.append_data(frame)
@@ -18,6 +21,7 @@ def save_video(frames: np.ndarray, output_path: str, fps: int = 16):
     except ImportError:
         try:
             import cv2
+
             h, w = frames.shape[1], frames.shape[2]
             fourcc = cv2.VideoWriter_fourcc(*"avc1")
             writer = cv2.VideoWriter(output_path, fourcc, fps, (w, h))
@@ -27,9 +31,11 @@ def save_video(frames: np.ndarray, output_path: str, fps: int = 16):
         except (ImportError, Exception):
             # Last resort: save as individual PNGs
             from PIL import Image
+
             out_dir = Path(output_path).parent / Path(output_path).stem
             out_dir.mkdir(parents=True, exist_ok=True)
             for i, frame in enumerate(frames):
                 Image.fromarray(frame).save(out_dir / f"frame_{i:04d}.png")
-            print(f"  (no video encoder available, saved {len(frames)} frames to {out_dir}/)")
-
+            print(
+                f"  (no video encoder available, saved {len(frames)} frames to {out_dir}/)"
+            )