Fix MyPy Errors

binkhq · Jun 29, 2023 · fa753b8 · fa753b8
1 parent 61ffd91
commit fa753b8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/soteria/requests_retry.py b/soteria/requests_retry.py
@@ -2,15 +2,15 @@
 
 import requests
 
-from requests.adapters import HTTPAdapter
-from requests.packages.urllib3.util.retry import Retry
+from requests.adapters import HTTPAdapter, Retry
+from typing import Optional
 
 
 def requests_retry_session(
     retries: int = 3,
     backoff_factor: float = 0.3,
     status_forcelist: t.Tuple = (500, 502, 504),
-    session: requests.Session = None,
+    session: Optional[requests.Session] = None,
 ) -> requests.Session:
     """
     Create a requests session with the given retry policy.