lint

src/airtrain/client.py

@@ -69,8 +69,8 @@ class AirtrainClient:
     def __init__(
         self, api_key: Optional[str] = None, base_url: Optional[str] = None
     ) -> None:
-        self._api_key: str = api_key or _find_api_key()
-        self._base_url: str = base_url or os.environ.get(
+        self._api_key: str = api_key or _find_api_key()  # type: ignore
+        self._base_url: str = base_url or os.environ.get(  # type: ignore
             "AIRTRAIN_API_URL", _DEFAULT_BASE_URL
         )
         if self._base_url.endswith("/"):
@@ -147,7 +147,7 @@ def _put_bytes(
             params=params,
             follow_redirects=False,
         )
-        if response.status_code not in (301, 302, 303, 307, 308):
+        if response.next_request is None:
             logger.error("Response text:\n%s", response.text)
             raise ServerError(f"Expected redirect but got: {response.status_code}")
 
@@ -204,7 +204,7 @@ def _handle_response(
             raise ServerError("Malformed server response.")
 
         if expect_json:
-            return response_json["data"]
+            return response_json["data"]  # type: ignore
         return None
 
     def _full_url(self, url_path: str) -> str: