diff --git a/pyicloud/base.py b/pyicloud/base.py index 572fdda6..f3eb8d24 100644 --- a/pyicloud/base.py +++ b/pyicloud/base.py @@ -8,7 +8,7 @@ from dataclasses import dataclass from os import chmod, environ, makedirs, path, umask from tempfile import gettempdir -from typing import Any, Dict, List, Mapping, Optional +from typing import Any, Dict, List, Mapping, Optional, cast from uuid import uuid1 import srp @@ -964,8 +964,9 @@ def request_2fa_code(self) -> bool: auth_endpoint=self._auth_endpoint, headers=self._get_auth_headers({"Accept": CONTENT_TYPE_JSON}), boot_context=self._current_hsa2_boot_context(), - user_agent=self.session.headers.get( - "User-Agent", _HEADERS["User-Agent"] + user_agent=cast( + str, + self.session.headers.get("User-Agent", _HEADERS["User-Agent"]), ), ) self._set_two_factor_delivery_state("trusted_device") diff --git a/requirements_test.txt b/requirements_test.txt index 41ec7f14..c1897899 100644 --- a/requirements_test.txt +++ b/requirements_test.txt @@ -7,3 +7,4 @@ pytest>=8.3.5 pytest-cov>=7.1.0 pytest-socket>=0.6.0 ruff>=0.9.9 +types-requests>=2.32.0