formatting and typing fix

landing-ai · Apr 23, 2024 · 669a88b · 669a88b
1 parent c5135c7
commit 669a88b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/examples/custom_tools/run_custom_tool.py b/examples/custom_tools/run_custom_tool.py
@@ -1,9 +1,9 @@
+from template_match import template_matching_with_rotation
+
 import vision_agent as va
 from vision_agent.image_utils import get_image_size, normalize_bbox
 from vision_agent.tools import Tool, register_tool
 
-from template_match import template_matching_with_rotation
-
 
 @register_tool
 class TemplateMatch(Tool):

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -905,7 +905,7 @@ def __call__(self, image: str) -> dict:
  raise ValueError(f"Request failed: {res.text}")
 
  data = res.json()
- output = {"labels": [], "bboxes": [], "scores": []}
+ output: Dict[str, List] = {"labels": [], "bboxes": [], "scores": []}
  for det in data[0]:
  output["labels"].append(det["text"])
  box = [