o $i{ @sddlZddlZddlZddlZddlmZmZmZmZe e Z   ddede deee de d e d ef d d Z   dde deee de d e d ef d dZdS)N)AnyCallableListOptional f descriptionmatch max_attempts max_backoff_sreturnc s$|dks Jd|dt|D]}z ||i|WSty}zgt||dup6tfdd|D} | rc|d|krctd||tdd } td |dd |d | d t | n"| rttd|d |d|dtdd|dd|dWYd}~qd}~wwdS)ahRetry a function with exponential backoff. Args: f: The function to retry. description: An imperative description of the function being retried. For example, "open the file". match: A list of strings to match in the exception message. If ``None``, any error is retried. max_attempts: The maximum number of attempts to retry. max_backoff_s: The maximum number of seconds to backoff. *args: Arguments to pass to the function. **kwargs: Keyword arguments to pass to the function. Returns: The result of the function. z%`max_attempts` must be positive. Got .Nc3s|]}|vVqdSN).0pattern exception_strrN/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/_common/retry.py +s z"call_with_retry..g?g333333?z Retrying z attempts to z after z seconds.z Failed to z attempts. Raising.zDid not find a match for z. Raising after z attempts.) range Exceptionstranyminrandomuniformloggerdebugtimesleep) rr r r r argskwargsie is_retryablebackoffrrrcall_with_retry s8   r*cs dtdtffdd }|S)z+Decorator-based version of call_with_retry.funcr cs"tfdd}|S)Ncstg|Ri|Sr)r*)r$r%)r r+r r r rrinnerJs z'retry..decorator..inner) functoolswraps)r+r,r r r r )r+r decoratorIszretry..decorator)r)r r r r r0rr/rretryAs r1)Nrr)r-loggingrr"typingrrrr getLogger__name__r rintr*r1rrrrsH   9