From 0a423741bd06681bda7eb129f1dc87d8710d1504 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Wed, 25 Sep 2024 11:00:02 -0700
Subject: [PATCH] fix flake8

---
 .../agent/vision_agent_coder_prompts.py        | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/vision_agent/agent/vision_agent_coder_prompts.py b/vision_agent/agent/vision_agent_coder_prompts.py
index 306cafb7..07f2c6e2 100644
--- a/vision_agent/agent/vision_agent_coder_prompts.py
+++ b/vision_agent/agent/vision_agent_coder_prompts.py
@@ -270,42 +270,42 @@ def get_counts(preds):
 
 **Input Code Snippet**:
 ```python
-from vision_agent.tools import load_image, owl_v2_image 
+from vision_agent.tools import load_image, owl_v2_image
 
 def check_helmets(image_path):
     image = load_image(image_path)
     # Detect people and helmets, filter out the lowest confidence helmet score of 0.15
     detections = owl_v2_image("person, helmet", image, box_threshold=0.15)
     height, width = image.shape[:2]
-    
+
     # Separate people and helmets
     people = [d for d in detections if d['label'] == 'person']
     helmets = [d for d in detections if d['label'] == 'helmet']
-    
+
     people_with_helmets = 0
     people_without_helmets = 0
-    
+
     for person in people:
         person_x = (person['bbox'][0] + person['bbox'][2]) / 2
         person_y = person['bbox'][1]  # Top of the bounding box
-        
+
         helmet_found = False
         for helmet in helmets:
             helmet_x = (helmet['bbox'][0] + helmet['bbox'][2]) / 2
             helmet_y = (helmet['bbox'][1] + helmet['bbox'][3]) / 2
-            
+
             # Check if the helmet is within 20 pixels of the person's head. Unnormalize
             # the coordinates so we can better compare them.
             if (abs((helmet_x - person_x) * width) < 20 and
                 -5 < ((helmet_y - person_y) * height) < 20):
                 helmet_found = True
                 break
-        
+
         if helmet_found:
             people_with_helmets += 1
         else:
             people_without_helmets += 1
-    
+
     return {{
         "people_with_helmets": people_with_helmets,
         "people_without_helmets": people_without_helmets
@@ -319,7 +319,7 @@ def check_helmets(image_path):
 3. **Pseudocode Creation**: Write down the steps you will follow in pseudocode.
 4. **Code Generation**: Translate your pseudocode into executable Python code.
     4.1. Take in the media path as an argument and load with either `load_image` or `extract_frames_and_timestamps`.
-    4.2. Coordinates are always returned normalized from `vision_agent.tools`. 
+    4.2. Coordinates are always returned normalized from `vision_agent.tools`.
     4.3. Do not create dummy input or functions, the code must be usable if the user provides new media.
     4.4. Use unnormalized coordinates when comparing bounding boxes.
 """