chore: Lower giveup log level for retried functions to warning (#26188)

This commit is contained in:
Jack Fragassi 2023-12-06 16:33:19 -08:00 committed by GitHub
parent c8e69e2f9d
commit bf5b18ccb1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions

View File

@ -15,6 +15,7 @@
# specific language governing permissions and limitations
# under the License.
import logging
from collections.abc import Generator
from typing import Any, Callable, Optional
@ -26,6 +27,7 @@ def retry_call(
*args: Any,
strategy: Callable[..., Generator[int, None, None]] = backoff.constant,
exception: type[Exception] = Exception,
giveup_log_level: int = logging.WARNING,
fargs: Optional[list[Any]] = None,
fkwargs: Optional[dict[str, Any]] = None,
**kwargs: Any
@ -33,6 +35,7 @@ def retry_call(
"""
Retry a given call.
"""
kwargs["giveup_log_level"] = giveup_log_level
decorated = backoff.on_exception(strategy, exception, *args, **kwargs)(func)
fargs = fargs or []
fkwargs = fkwargs or {}