o -wi @sddlmZddlmZddlmZmZmZm Z m Z ej j eej j eej jeej je ej je iZddeDZGdddZd S) )Optional)wandb_internal_pb2)AuthenticationError CommErrorErrorUnsupportedError UsageErrorcCsi|]\}}||qSr ).0kvr r N/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/errors/util.py src@sBeZdZdZedejdeefddZ e deddfdd Z d S) ProtobufErrorHandlerz6Converts protobuf errors to exceptions and vice versa.errorreturncCs0|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_exceptionexcz pb.ErrorInfocCsLt|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) __name__ __module__ __qualname____doc__ staticmethodrrrrr classmethodr#r r r rrs rN)typingr wandb.protorrrrrrr rUNKNOWN COMMUNICATIONAUTHENTICATIONUSAGE UNSUPPORTEDritemsrrr r r rs