From 34580e1257aa903848387198ca73c043e381fa93 Mon Sep 17 00:00:00 2001
From: Dillon Laird <dillonalaird@gmail.com>
Date: Thu, 18 Apr 2024 11:35:55 -0700
Subject: [PATCH] formatting fix

---
 vision_agent/image_utils.py    | 4 +++-
 vision_agent/tools/__init__.py | 2 +-
 vision_agent/tools/tools.py    | 7 +++++--
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/vision_agent/image_utils.py b/vision_agent/image_utils.py
index 3a34fe93..f36a2033 100644
--- a/vision_agent/image_utils.py
+++ b/vision_agent/image_utils.py
@@ -103,7 +103,9 @@ def overlay_bboxes(
     elif isinstance(image, np.ndarray):
         image = Image.fromarray(image)
 
-    color = {label: COLORS[i % len(COLORS)] for i, label in enumerate(set(bboxes["labels"]))}
+    color = {
+        label: COLORS[i % len(COLORS)] for i, label in enumerate(set(bboxes["labels"]))
+    }
 
     width, height = image.size
     fontsize = max(12, int(min(width, height) / 40))
diff --git a/vision_agent/tools/__init__.py b/vision_agent/tools/__init__.py
index 2e93e340..1c1c6e73 100644
--- a/vision_agent/tools/__init__.py
+++ b/vision_agent/tools/__init__.py
@@ -6,10 +6,10 @@
     BboxIoU,
     BoxDistance,
     Crop,
+    DINOv,
     ExtractFrames,
     GroundingDINO,
     GroundingSAM,
-    DINOv,
     ImageCaption,
     SegArea,
     SegIoU,
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
index 5461999b..f53ee372 100644
--- a/vision_agent/tools/tools.py
+++ b/vision_agent/tools/tools.py
@@ -442,10 +442,13 @@ def __call__(
         }
         data: Dict[str, Any] = _send_inference_request(request_data, "dinov")
         if "bboxes" in data:
-            data["bboxes"] = [normalize_bbox(box, data["mask_shape"]) for box in data["bboxes"]]
+            data["bboxes"] = [
+                normalize_bbox(box, data["mask_shape"]) for box in data["bboxes"]
+            ]
         if "masks" in data:
             data["masks"] = [
-                rle_decode(mask_rle=mask, shape=data["mask_shape"]) for mask in data["masks"]
+                rle_decode(mask_rle=mask, shape=data["mask_shape"])
+                for mask in data["masks"]
             ]
         data["labels"] = ["visual prompt" for _ in range(len(data["masks"]))]
         return data