o xi @sddlmZddlmZddlmZmZmZm Z m Z ej j eej j eej jeej je ej je iZddeDZGdddZd S) ) annotations)wandb_internal_pb2)AuthenticationError CommErrorErrorUnsupportedError UsageErrorcCsi|]\}}||qSr ).0kvr r E/home/ubuntu/.local/lib/python3.10/site-packages/wandb/errors/util.py src@s,eZdZdZed ddZedd d Zd S)ProtobufErrorHandlerz6Converts protobuf errors to exceptions and vice versa.error pb.ErrorInforeturn Error | NonecCs0|sdS|jtvrt|j|jSt|jS)zConvert a protobuf error to an exception. Args: error: The protobuf error to convert. Returns: The corresponding exception. N)SerializeToStringcodeto_exception_mapmessager)rr r r to_exceptions    z!ProtobufErrorHandler.to_exceptionexcrcCsLt|ts tdd}t|jD] }|tvrt|}nqtj|t|dS)zConvert an wandb error to a protobuf error message. Args: exc: The exception to convert. Returns: The corresponding protobuf error message. z,exc must be a subclass of wandb.errors.ErrorN)rr) isinstancer TypeErrortype__mro__from_exception_mappb ErrorInfostr)clsrrsubclassr r rfrom_exception's z#ProtobufErrorHandler.from_exceptionN)rrrr)rrrr)__name__ __module__ __qualname____doc__ staticmethodr classmethodr%r r r rrs  rN) __future__r wandb.protorr rrrrr r!UNKNOWN COMMUNICATIONAUTHENTICATIONUSAGE UNSUPPORTEDritemsrrr r r rs