Vh~ddlZddlmZddlmZGddeZejdeefdZ Gdd Z y) N)Hashable)fieldsc>eZdZUdZeed<edZdefdZ dZ y) _UnionTag)_clsrcFt|}t|drJ||_|S)Nr)rhasattrr)tclstags I/home/dcms/DCMS/lib/python3.12/site-packages/torch/_export/serde/union.pycreatez_UnionTag.create s'l3''' returnct|tsJt|}|t|jvs,J|d|jdt|jt||k(S)Nz is not a valid tag for z. Available tags: ) isinstancestr_get_field_namesr)selfcmpothers r __eq__z_UnionTag.__eq__s|#s###C( II   hW,TYYK7IJZ[_[d[dJeIf g h 4yE!!rc*tt|SN)hashrrs r __hash__z_UnionTag.__hash__sCIrN) __name__ __module__ __qualname__ __slots__r__annotations__ staticmethodrboolrrrr rrs/I N "T"rrrcRt|Dchc]}|jc}Scc}wr)rname)r fs r rrs"3K (qAFF (( (s$creZdZUeed<edZdZede fdZ edZ fdZ dZ d ZxZS) _Union_typec t|dk(sJ|diit|Dcic]}|jdc}|}tj t t |j||_|Scc}w)Nr%) lenrr'rrnextiterkeysr+)r kwargsr(objs r rz _Union.create&sq6{aFEfSk: :EfEF$$T$v{{}*=%>D  ;sA<c>tdt|DrJy)Nc38K|]}|jdvyw))typer+rvalueN)r').0r(s r z'_Union.__post_init__...s\!qvv!EE\s)anyrrs r __post_init__z_Union.__post_init__-s\vVZ|\\\\\rrc |jS#t$r(}tdt|jd|d}~wwxYw)Nz Please use z&.create to instantiate the union type.)r+AttributeError RuntimeErrorr6r)res r r6z _Union.type0sI ::  d4j1122XY  s  ?#:?c.t||jSr)getattrr6rs r r7z _Union.value9stTYY''rct||}|4|tt|vr||jk7rt d|d|S)NzField z is not set.)super__getattribute__rr6r=)rr'attr __class__s r rDz_Union.__getattribute__=sLw'- rVsI$. )SX))$P$Pr