o }oi@sUddlmZddlZddlZddlmZddlmZee Z er.ddlm Z ddl m Z GdddZerIdd lmZeed Zd ed <dSdS) ) annotationsN) TYPE_CHECKING)ArtifactNotFound)BinaryIO) ArtifactStorec@sNeZdZdZdddddd!ddZd"ddZd#ddZd$ddZd%ddZd S)&BackoffaAn artifact store's middleware for exponential backoff. Example: .. code-block:: python import optuna from optuna.artifacts import upload_artifact from optuna.artifacts import Boto3ArtifactStore from optuna.artifacts import Backoff artifact_store = Backoff(Boto3ArtifactStore("my-bucket")) def objective(trial: optuna.Trial) -> float: ... = trial.suggest_float("x", -10, 10) file_path = generate_example(...) upload_artifact( artifact_store=artifact_store, file_path=file_path, study_or_trial=trial, ) return ... g?) max_retries multiplier min_delay max_delaybackendrr intr floatr rreturnNonecCsR||_|dks J|dksJ|dksJ||ksJ||_||_||_||_dS)Nr)_backend _max_retries _multiplier _min_delay _max_delay)selfrr r r rrM/home/ubuntu/.local/lib/python3.10/site-packages/optuna/artifacts/_backoff.py__init__-s     zBackoff.__init__n_retrycCst|j|j||jS)N)minrrr)rrrrr_get_sleep_secsBszBackoff._get_sleep_secs artifact_idstrrc Cst|jD]B}z |j|WStyty>}z||jdkr'tjd|d||dWYd}~nd}~wwt | |qJd)NFailed to open artifact= n_retry=exc_infoFzmust not reach here) rangerr open_readerr Exception_loggererrortimesleeprrr ierrrr(Es &zBackoff.open_reader content_bodyc Cst|jD]I}z |j||WdStyty@}z||jdkr)tjd|d||dWYd}~nd}~ww|dt | |qdS)Nr"r#r$r%r) r'rrwriterr)r*r+seekr,r-r)rr r1r/r0rrrr2Ss & z Backoff.writec Cst|jD]>}z|j|Wn+tyty:}z||jdkr&tjd||dWYd}~nd}~wwt | |qdS)Nr"zFailed to delete artifact=r%) r'rrremoverr)r*r+r,r-rr.rrrr4bs  zBackoff.removeN) rrr rr rr rrrrr)rrrr)r r!rr)r r!r1rrr)r r!rr) __name__ __module__ __qualname____doc__rrr(r2r4rrrrrs   r)FileSystemArtifactStore.r_) __future__rloggingr,typingroptuna.artifacts.exceptionsr getLoggerr5r*roptuna.artifacts._protocolrroptuna.artifactsr9r;__annotations__rrrrs     ]