fix linting

landing-ai · Apr 27, 2024 · cd9932c · cd9932c
1 parent 8766f83
commit cd9932c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -9,7 +9,7 @@
 import requests
 from PIL import Image
 from PIL.Image import Image as ImageType
-from scipy.spatial import distance
+from scipy.spatial import distance # type: ignore
 
 from vision_agent.image_utils import (
  b64_to_pil,
@@ -603,12 +603,12 @@ def __call__(
  """
  image_size = get_image_size(image)
  bbox = prompt["bbox"]
- prompt = ", ".join(map(str, denormalize_bbox(bbox, image_size)))
+ bbox_str = ", ".join(map(str, denormalize_bbox(bbox, image_size)))
  image_b64 = convert_to_b64(image)
 
  data = {
  "image": image_b64,
- "prompt": prompt,
+ "prompt": bbox_str,
  "tool": "few_shot_counting",
  }
  resp_data = _send_inference_request(data, "tools")