fix mypy errors

landing-ai · Aug 28, 2024 · 8ee040e · 8ee040e
1 parent 10b3151
commit 8ee040e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/integ/test_tools.py b/tests/integ/test_tools.py
@@ -198,7 +198,7 @@ def test_countgd_counting() -> None:
 def test_countgd_example_based_counting() -> None:
  img = ski.data.coins()
  result = countgd_example_based_counting(
- visual_prompt=[[85, 106, 122, 145]],
+ visual_prompts=[[85, 106, 122, 145]],
  image=img,
  )
  assert result["count"] == 24

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -541,7 +541,7 @@ def countgd_counting(
  "box_threshold": box_threshold,
  "function_name": "countgd_counting",
  }
- data: Dict[str, Any] = send_inference_request(
+ data: List[Dict[str, Any]] = send_inference_request(
  payload, "countgd_counting", files=files, v2=True
  )
  return data
@@ -588,7 +588,7 @@ def countgd_example_based_counting(
  "box_threshold": box_threshold,
  "function_name": "countgd_example_based_counting",
  }
- data: Dict[str, Any] = send_inference_request(
+ data: List[Dict[str, Any]] = send_inference_request(
  payload, "countgd_example_based_counting", files=files, v2=True
  )
  return data