added type hint to check_timeout_on_yield

googleapis · Aug 15, 2023 · 41c7868 · 41c7868
1 parent c9ef1d5
commit 41c7868
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/google/api_core/retry_streaming.py b/google/api_core/retry_streaming.py
@@ -135,7 +135,7 @@ def __init__(
         exception_factory: Optional[
             Callable[[list[Exception], bool, float], tuple[Exception, Exception | None]]
         ] = None,
-        check_timeout_on_yield=False,
+        check_timeout_on_yield: bool = False,
     ):
         """
         Args:

diff --git a/google/api_core/retry_streaming_async.py b/google/api_core/retry_streaming_async.py
@@ -32,7 +32,6 @@
 import time
 from functools import partial
 
-from google.api_core import exceptions
 from google.api_core.retry_streaming import _build_timeout_error
 
 _LOGGER = logging.getLogger(__name__)
@@ -123,7 +122,7 @@ def __init__(
         exception_factory: Optional[
             Callable[[list[Exception], bool, float], tuple[Exception, Exception | None]]
         ] = None,
-        check_timeout_on_yield=False,
+        check_timeout_on_yield: bool = False,
     ):
         """
         Args: