From 41c7868f6557df4c824d8b571e91cfdcce036aff Mon Sep 17 00:00:00 2001 From: Daniel Sanche Date: Tue, 15 Aug 2023 09:38:40 -0700 Subject: [PATCH] added type hint to check_timeout_on_yield --- google/api_core/retry_streaming.py | 2 +- google/api_core/retry_streaming_async.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/google/api_core/retry_streaming.py b/google/api_core/retry_streaming.py index fad89b25..265a1525 100644 --- 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 index dd90dd66..5f00fa5d 100644 --- 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: