code: def retry_func_accept_retry_state(retry_func):
      
      if not six.callable(retry_func):
          return retry_func
  
      if func_takes_retry_state(retry_func):
          return retry_func
  
      @_utils.wraps(retry_func)
      def wrapped_retry_func(retry_state):
          warn_about_non_retry_state_deprecation(
              'retry', retry_func, stacklevel=4)
          return retry_func(retry_state.outcome)
      return wrapped_retry_func
docs: Wrap "retry" function to accept "retry_state" parameter.