From a4547b569a45ed631bc30e359003368062ee6bb9 Mon Sep 17 00:00:00 2001 From: Jack Fragassi Date: Wed, 6 Dec 2023 16:33:19 -0800 Subject: [PATCH] chore: Lower giveup log level for retried functions to warning (#26188) --- superset/utils/retries.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/superset/utils/retries.py b/superset/utils/retries.py index 8a1e6b95eadcb..3af821362de93 100644 --- a/superset/utils/retries.py +++ b/superset/utils/retries.py @@ -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 {}