fix typing

Author: sleepyStick

pymongo/asynchronous/client_session.py

@@ -715,12 +715,13 @@ async def callback(session, custom_arg, custom_kwarg=None):
         """
         start_time = time.monotonic()
         retry = 0
-        last_error = None
+        last_error: Optional[BaseException] = None
         while True:
             if retry:  # Implement exponential backoff on retry.
                 jitter = random.random()  # noqa: S311
                 backoff = jitter * min(_BACKOFF_INITIAL * (1.25**retry), _BACKOFF_MAX)
                 if _would_exceed_time_limit(start_time, backoff):
+                    assert last_error is not None
                     raise last_error
                 await asyncio.sleep(backoff)
             retry += 1

pymongo/synchronous/client_session.py

@@ -713,12 +713,13 @@ def callback(session, custom_arg, custom_kwarg=None):
         """
         start_time = time.monotonic()
         retry = 0
-        last_error = None
+        last_error: Optional[BaseException] = None
         while True:
             if retry:  # Implement exponential backoff on retry.
                 jitter = random.random()  # noqa: S311
                 backoff = jitter * min(_BACKOFF_INITIAL * (1.25**retry), _BACKOFF_MAX)
                 if _would_exceed_time_limit(start_time, backoff):
+                    assert last_error is not None
                     raise last_error
                 time.sleep(backoff)
             retry += 1