From 98e8bc84f6c585e0306478b5a4b14d72bba4d4a0 Mon Sep 17 00:00:00 2001 From: wuyiqunLu Date: Sat, 14 Sep 2024 13:37:21 +0800 Subject: [PATCH] fix lint --- vision_agent/agent/vision_agent.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/vision_agent/agent/vision_agent.py b/vision_agent/agent/vision_agent.py index 216419d5..40195246 100644 --- a/vision_agent/agent/vision_agent.py +++ b/vision_agent/agent/vision_agent.py @@ -232,8 +232,10 @@ def chat_with_code( if file_path.lower().endswith( ".mp4" ) or file_path.lower().endswith(".mov"): - ndarray = extract_frames(media) - save_video([frame for frame, _ in ndarray], file_path) + video_frames = extract_frames(media) + save_video( + [frame for frame, _ in video_frames], file_path + ) else: ndarray = load_image(media) save_image(ndarray, file_path)