Merge pull request #2316 from VWS-Python/common-io-type

Use consistent way of setting image type

Author: adamtheturtle

src/vws/query.py

@@ -3,9 +3,9 @@
 """
 
 import datetime
+import io
 import json
 from http import HTTPMethod, HTTPStatus
-from io import BufferedRandom, BytesIO
 from typing import Any
 from urllib.parse import urljoin
 
@@ -29,7 +29,7 @@
 from vws.reports import QueryResult, TargetData
 
 
-def _get_image_data(image: BytesIO | BufferedRandom) -> bytes:
+def _get_image_data(image: io.BytesIO | io.BufferedRandom) -> bytes:
     """Get the data of an image file."""
     original_tell = image.tell()
     image.seek(0)
@@ -61,7 +61,7 @@ def __init__(
 
     def query(
         self,
-        image: BytesIO | BufferedRandom,
+        image: io.BytesIO | io.BufferedRandom,
         max_num_results: int = 1,
         include_target_data: CloudRecoIncludeTargetData = (
             CloudRecoIncludeTargetData.TOP

src/vws/vws.py

@@ -8,7 +8,7 @@
 import time
 from datetime import date
 from http import HTTPMethod, HTTPStatus
-from io import BufferedRandom, BytesIO
+from io import BufferedRandom
 from urllib.parse import urljoin
 
 import requests
@@ -49,7 +49,7 @@
 from .exceptions.response import Response
 
 
-def _get_image_data(image: BytesIO | BufferedRandom) -> bytes:
+def _get_image_data(image: io.BytesIO | BufferedRandom) -> bytes:
     """Get the data of an image file."""
     original_tell = image.tell()
     image.seek(0)
@@ -234,7 +234,7 @@ def add_target(
         self,
         name: str,
         width: float,
-        image: BytesIO | BufferedRandom,
+        image: io.BytesIO | io.BufferedRandom,
         application_metadata: str | None,
         *,
         active_flag: bool,