Fix typing errors

landing-ai · Mar 11, 2024 · dfdb792 · dfdb792
1 parent e4f3b7e
commit dfdb792
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/vision_agent/image_utils.py b/vision_agent/image_utils.py
@@ -7,7 +7,7 @@
 from PIL import Image
 
 
-def b64_to_pil(b64_str: str) -> Image:
+def b64_to_pil(b64_str: str) -> Image.Image:
  # , can't be encoded in b64 data so must be part of prefix
  if "," in b64_str:
  b64_str = b64_str.split(",")[1]

diff --git a/vision_agent/tools/tools.py b/vision_agent/tools/tools.py
@@ -60,8 +60,8 @@ def __call__(self, image: Union[str, Path, ImageType]) -> List[Dict]:
  )
  resp_json: Dict[str, Any] = res.json()
  if resp_json["statusCode"] != 200:
- _LOGGER.error(f"Request failed: {resp_json['data']}")
- return cast(str, resp_json["data"])
+ _LOGGER.error(f"Request failed: {resp_json}")
+ return cast(List[Dict], resp_json["data"])
 
 
 class GroundingSAM(ImageTool):