From a19c6f8238f027724260d0aba9be8353bff0f9a2 Mon Sep 17 00:00:00 2001 From: erjia Date: Wed, 15 Feb 2023 12:18:17 -0800 Subject: [PATCH] Fix mypy after updates from `types_requests` (#1018) Summary: Fix the mypy Error after `types_requests` is released to `2.28.11.13` Pull Request resolved: https://github.com/pytorch/data/pull/1018 Reviewed By: NivekT Differential Revision: D43309767 Pulled By: ejguan fbshipit-source-id: 1990ec809156c961ebf32d6f3b16b67071cc1303 --- torchdata/datapipes/iter/load/online.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torchdata/datapipes/iter/load/online.py b/torchdata/datapipes/iter/load/online.py index 270a86683..dbbe8c916 100644 --- a/torchdata/datapipes/iter/load/online.py +++ b/torchdata/datapipes/iter/load/online.py @@ -36,7 +36,7 @@ def _get_response_from_http( ) -> Tuple[str, StreamWrapper]: with requests.Session() as session: proxies = _get_proxies() - r = session.get(url, timeout=timeout, proxies=proxies, stream=True, **query_params) # type: ignore[arg-type] + r = session.get(url, timeout=timeout, proxies=proxies, stream=True, **query_params) # type: ignore[attr-defined] r.raise_for_status() return url, StreamWrapper(r.raw) @@ -105,7 +105,7 @@ def _get_response_from_google_drive( confirm_token = None with requests.Session() as session: - response = session.get(url, timeout=timeout, stream=True, **query_params) # type: ignore[arg-type] + response = session.get(url, timeout=timeout, stream=True, **query_params) # type: ignore[attr-defined] response.raise_for_status() for k, v in response.cookies.items(): @@ -122,7 +122,7 @@ def _get_response_from_google_drive( if confirm_token: url = url + "&confirm=" + confirm_token - response = session.get(url, timeout=timeout, stream=True, **query_params) # type: ignore[arg-type] + response = session.get(url, timeout=timeout, stream=True, **query_params) # type: ignore[attr-defined] response.raise_for_status() if "content-disposition" not in response.headers: