From 1b9418c5bb9208501df3a0bb197eb9eaa1d285f2 Mon Sep 17 00:00:00 2001
From: Zhichao <Zhichao.lin@landing.ai>
Date: Fri, 6 Sep 2024 21:48:33 +0800
Subject: [PATCH] minor

---
 vision_agent/utils/video.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vision_agent/utils/video.py b/vision_agent/utils/video.py
index 8aefb754..51774279 100644
--- 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"