o xi @sUdZddlmZddlZddlmZmZddlm Z m Z ddl m Z ddlmZmZe rKddlmZdd lmZdd lmZdd lmZdd lmZiZd ed<GdddeZdS)zStorage policy.) annotationsN)ABCabstractmethod) TYPE_CHECKINGAny)Api) FilePathStrURIStr) StepPrepare)StoragePolicyConfig)Artifact)ArtifactManifestEntry) ProgressFnzdict[str, type[StoragePolicy]]_POLICY_REGISTRYcseZdZUdZded<d;fdd ZedddZ ed?ddZ e  d@dAd!d"Z e dBdCd+d,Z e  - dDdEd4d5Ze 6 dFdGd9d:ZZS)H StoragePolicyNzInternalApi | None_apikwargsrreturnNonec s"tjdi||t|<dS)N)super__init_subclass__rname)clsr __class__rV/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/artifacts/storage_policy.pyrszStoragePolicy.__init_subclass__rstrtype[StoragePolicy]cCs t|}r |Std|)NzFailed to find storage policy )rget ValueError)rrpolicyrrrlookup_by_nameszStoragePolicy.lookup_by_namecCtNNotImplementedError)rrrrr$zStoragePolicy.nameconfigr cCr#r$r%)rr(rrr from_config)r'zStoragePolicy.from_configdict[str, Any]cCr#r$r%)selfrrrr(.szStoragePolicy.configartifactr manifest_entryr dest_path str | Noneexecutor"concurrent.futures.Executor | NonercCr#r$r%)r+r,r-r.r0rrr load_file2szStoragePolicy.load_file artifact_idartifact_manifest_identrypreparerr progress_callbackProgressFn | NoneboolcCr#r$r%)r+r3r4r5r6r7rrr store_file< zStoragePolicy.store_fileTpathURIStr | FilePathStrchecksum max_objects int | Nonelist[ArtifactManifestEntry]cCr#r$r%)r+r,r<rr>r?rrrstore_referenceGr;zStoragePolicy.store_referenceFlocalFilePathStr | URIStrcCr#r$r%)r+r-rCr.rrrload_referenceRszStoragePolicy.load_reference)rrrr)rrrr)rr)r(r rr)rr*)NN) r,r r-r r.r/r0r1rrr$) r3rr4rr5r r6r r7r8rr9)NTN) r,r r<r=rr/r>r9r?r@rrA)FN)r-r rCr9r.r/rrD)__name__ __module__ __qualname__r__annotations__r classmethodr"rrr)r(r2r:rBrE __classcell__rrrrrs:      r)__doc__ __future__rconcurrent.futures concurrentabcrrtypingrrwandb.sdk.internal.internal_apir InternalApiwandb.sdk.lib.pathsrr wandb.filesync.step_preparer #wandb.sdk.artifacts._models.storager wandb.sdk.artifacts.artifactr +wandb.sdk.artifacts.artifact_manifest_entryr wandb.sdk.internal.progressrrrIrrrrrs