o }oi @sUddlmZddlmZddlmZddlmZddlm Z ddl m Z er-ddlm Z e Z ddlZWdn1s?wYed Gd d d Zeradd lmZed Zded<dSdS)) annotations)BytesIO) TYPE_CHECKING)experimental_class) try_import)ArtifactNotFound)BinaryIONz3.4.0c@s<eZdZdZ ddd d Zdd dZdddZdddZdS)GCSArtifactStorea|An artifact backend for Google Cloud Storage (GCS). Args: bucket_name: The name of the bucket to store artifacts. client: A google-cloud-storage ``Client`` to use for storage operations. If not specified, a new client will be created with default settings. Example: .. code-block:: python import optuna from optuna.artifacts import GCSArtifactStore, upload_artifact artifact_backend = GCSArtifactStore("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 ... Before running this code, you will have to install ``gcloud`` and run .. code-block:: bash gcloud auth application-default login so that the Cloud Storage library can automatically find the credential. N bucket_namestrclient"google.cloud.storage.Client | NonereturnNonecCs2t||_|ptjj|_|j||_ dSN) _importscheckr googlecloudstorageClientr bucket bucket_obj)selfr r rI/home/ubuntu/.local/lib/python3.10/site-packages/optuna/artifacts/_gcs.py__init__;szGCSArtifactStore.__init__ artifact_id 'BinaryIO'cCs<|j|}|durtd|jd|d|}t|S)NzArtifact storage with bucket: z, artifact_id: z was not found)rget_blobrr download_as_bytesr)rrblobbodyrrr open_readerEs zGCSArtifactStore.open_reader content_bodycCs"|j|}|}||dSr)rr!readupload_from_string)rrr$r!datarrrwriteQs zGCSArtifactStore.writecCs|j|dSr)r delete_blob)rrrrrremoveVszGCSArtifactStore.remover)r r r r rr)rr rr)rr r$rrr)rr rr)__name__ __module__ __qualname____doc__rr#r(r*rrrrr s* r ) ArtifactStorer/_) __future__riortypingroptuna._experimentalroptuna._importsroptuna.artifacts.exceptionsrrrgoogle.cloud.storagerr optuna.artifacts._protocolr/r1__annotations__rrrrs"       G