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 {}