Type check urllib3

Author: adamtheturtle

pyproject.toml

@@ -258,6 +258,7 @@ dev = [
     "sphinxcontrib-spelling==8.0.0",
     "sybil==5.0.0",
     "types-requests==2.28.11.16",
+    "types-urllib3==1.26.25.9",
     "vulture==2.7",
 ]
 

src/vws/include_target_data.py

@@ -2,15 +2,16 @@
 Tools for managing ``CloudRecoService.query``'s ``include_target_data``.
 """
 
-from enum import Enum
 
+from enum import StrEnum, auto
 
-class CloudRecoIncludeTargetData(Enum):
+
+class CloudRecoIncludeTargetData(StrEnum):
     """
     Options for the ``include_target_data`` parameter of
     ``CloudRecoService.query``.
     """
 
-    TOP = "top"
-    NONE = "none"
-    ALL = "all"
+    TOP = auto()
+    NONE = auto()
+    ALL = auto()

src/vws/query.py

@@ -6,7 +6,7 @@
 
 import datetime
 from http import HTTPStatus
-from typing import BinaryIO
+from typing import Any, BinaryIO
 from urllib.parse import urljoin
 
 import requests
@@ -102,7 +102,7 @@ def query(
             An ordered list of target details of matching targets.
         """
         image_content = _get_image_data(image=image)
-        body = {
+        body: dict[str, Any] = {
             "image": ("image.jpeg", image_content, "image/jpeg"),
             "max_num_results": (None, int(max_num_results), "text/plain"),
             "include_target_data": (
@@ -113,12 +113,7 @@ def query(
         }
         date = rfc_1123_date()
         request_path = "/v1/query"
-        (
-            content,
-            content_type_header,
-        ) = encode_multipart_formdata(  # type:ignore[no-untyped-call]
-            fields=body,
-        )
+        content, content_type_header = encode_multipart_formdata(fields=body)
         method = "POST"
 
         authorization_string = authorization_header(