U zh@sdddlZddlmZddlmZmZGdddeZejddeeddd Z Gd d d Z dS) Nfields)HashableSetc@s8eZdZUeed<eddZedddZddZ d S) _UnionTag_clscCs t|}t|drt||_|S)Nr)rhasattrAssertionErrorr)tclstagr K/var/www/html/venv/lib/python3.8/site-packages/torch/_export/serde/union.pycreate sz_UnionTag.createreturncCsPt|tstt|}|t|jksDt|d|jdt|jt||kS)Nz is not a valid tag for z. Available tags: ) isinstancestrr _get_field_namesr)selfcmpotherr r r__eq__sz_UnionTag.__eq__cCs tt|SN)hashrrr r r__hash__sz_UnionTag.__hash__N) __name__ __module__ __qualname__r__annotations__ staticmethodrboolrrr r r rrs  r)maxsizercCsddt|DS)NcSsh|] }|jqSr name.0fr r r sz#_get_field_names..r)r r r rrsrcsheZdZUeed<eddZddZee dddZ ed d Z fd d Z d dZ ddZZS)_Union_typecKsJt|dkst|fddt|D|}ttt|||_|S)NcSsi|] }|jdqSrr$r&r r r (sz!_Union.create..) lenr rrrnextiterkeysr+)r kwargsobjr r rr%sz _Union.createcCstddt|DrtdS)Ncss|]}|jdkVqdS))typer+rvalueNr$r&r r r -sz'_Union.__post_init__..)anyrr rr r r __post_init__,sz_Union.__post_init__rc CsHz|jWStk rB}ztdt|jd|W5d}~XYnXdS)Nz Please use z&.create to instantiate the union type.)r+AttributeError RuntimeErrorr4r)rer r rr4/sz _Union.typecCs t||jSr)getattrr4rr r rr58sz _Union.valuecsBt|}|dkr>|tt|kr>||jkr>td|d|S)NzField z is not set.)super__getattribute__rr4r9)rr%attr __class__r rr><s "z_Union.__getattribute__cCs|Sr)__repr__rr r r__str__Bsz_Union.__str__cCs&t|jd|jdt||jdS)N(=))r4rr<rr r rrBEsz_Union.__repr__)rrrrr classmethodrr8propertyrr4r5r>rCrB __classcell__r r r@rr*"s    r*) functools dataclassesrtypingrrrr lru_cacherr*r r r rs