o -wis @sFddlmZddlmZddlmZddlmZeGdddZdS)) annotations) dataclass)Union)parsec@sJeZdZUdZded< ded< dZded< dd Zedd d ZdS) RunMomentzA moment in a run. Defines a branching point in a finished run to fork or resume from. A run moment is identified by a run ID and a metric value. Currently, only the metric '_step' is supported. strrunzUnion[int, float]value_stepmetriccCs`|jdkrtd|jdt|jttfstd|jdt|jts.td|jddS)Nr z+Only the metric '_step' is supported, got 'z'.z-Only int or float values are supported, got 'z*Only string run names are supported, got ') r ValueError isinstancer intfloat TypeErrorrr)selfrU/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/wandb/sdk/lib/run_moment.py __post_init__s    zRunMoment.__post_init__urireturnc Csd|}td|d}zt|}Wnty"}z||d}~ww|jdkr*||js/||j}|js;|js;|jr=|t|j }t |dkrK|t | d}|dkrY|||d} z| rht| nt| } Wnty|}z||d}~ww|||| dS) Nz runmoment://z*Could not parse passed run moment string 'z', expected format '?='. Currently, only the metric '_step' is supported. Example: 'ans3bsax?_step=123'. runmomentrr )rr r )r rurlparseschemenetlocpathparamsfragmentparse_qsquerylenlistkeysisdigitrr) clsrparsable parse_errparsederr r r num_valuerrrfrom_uri*s>     zRunMoment.from_uriN)rrrr) __name__ __module__ __qualname____doc____annotations__r r classmethodr+rrrrrs   rN) __future__r dataclassesrtypingrurllibrrrrrrs