Регистрация упорства python retry_state с помощью регистратора из внешней области

Question

У меня есть модуль, который включает служебную функцию с тегом повторной попытки упорства.

      from tenacity import retry, stop_after_attempt, wait_random

def log_attempt_number(logger, retry_state):
  logger.info(f"Attempt number {retry_state}")
  logger.info("Logger is not recognized here unless we instantiate in this file")

@retry(stop=stop_after_attempt(3), wait=wait_random(min=1, max=3), before=log_attempt_number)
def retry_function(logger, driver):
  logger.info("Outer scope logger works fine here")
  source = driver.page_source
  return source

Посколькуretry_functionиспользуется в других сценариях, я хочу иметь возможность использовать существующие регистраторы этих сценариев для вывода, а не создавать экземпляр регистратора в этом модуле. Однако я не могу понять, как подключить регистратор внешней области для использования вlog_attempt_numberфункция, так как она вызывается только внутри Tenacity@retryкоторый, кажется, на самом деле не принимает аргументы.