minor

landing-ai · Sep 6, 2024 · 1b9418c · 1b9418c
1 parent f35ce7b
commit 1b9418c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vision_agent/utils/video.py b/vision_agent/utils/video.py
@@ -58,7 +58,7 @@ def video_writer(
  filename = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
  container = av.open(filename, mode="w")
  stream = container.add_stream("h264", rate=fps)
- height, width = frames[0].shape[:1]
+ height, width = frames[0].shape[:2]
  stream.height = height - (height % 2)
  stream.width = width - (width % 2)
  stream.pix_fmt = "yuv420p"