fix lint

landing-ai · Sep 14, 2024 · 98e8bc8 · 98e8bc8
1 parent 5509c3d
commit 98e8bc8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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)