U ‰d9ã@s€ddlZddlmZddlmZeGdd„dƒƒZdd„ZeGdd „d ƒƒZdd d œeedd œd d„Zeee d œdd„Z dS)éN)ÚIterator)Ú dataclassc@seZdZUeed<eed<dS)ÚMetaInitErrorInfoÚ mode_nameÚmode_class_nameN)Ú__name__Ú __module__Ú __qualname__ÚstrÚ__annotations__©r r ú;/tmp/pip-unpacked-wheel-ua33x9lu/torch/utils/_mode_utils.pyrs rcs*tƒ‰t ˆ¡ˆdœ‡‡‡fdd„ ƒ}|S)N©Úinnercs8|ˆkr"tdˆj›dˆj›dƒ‚||_ˆ|f|ž|ŽS)Nz:missing inner keyword argument; instead of constructing a z( directly, pass the constructor to push_Ú_mode)Ú TypeErrorrrr)ÚselfrÚargsÚkwargs©ÚfÚmeta_init_error_infoZundefr r Úwrappeds ÿz_wrap_init..wrapped)ÚobjectÚ functoolsÚwraps)rrrr rr Ú _wrap_initsrc@s>eZdZUeed<eed<eed<dd„Zdd„Zdd „Zd S) Ú _ModeInforÚ mode_classÚbase_mode_classcCs|jjS)N)rr©rr r r r/sz_ModeInfo.mode_class_namecCs tƒ‚dS)z+gets the current mode for this type of modeN©ÚNotImplementedErrorr r r r Úget_mode2sz_ModeInfo.get_modecCs tƒ‚dS)zÄ set mode to for this type of mode. Note that no checks are done on this, it's the unsafe version where checks are assumed to have been already done by the helper function Nr!)rÚmoder r r Úset_mode6sz_ModeInfo.set_modeN) rrr r r Útyperr#r%r r r r r)s rF)ÚreplaceÚignore_preexisting)Ú mode_infoÚreturnc cs|dksHt||jƒsHt|tƒr*t||jƒrHtd| ¡›dt|ƒ›dƒ‚| ¡}||krbdVdS|dk r¾|s¾||k r¾t||jƒr’d|j›d}nd| ¡›d}td|j›d |›d |›ƒ‚d |jd }t||ƒsìtd |j›d |›ƒ‚|  |¡z dVW5|  |¡XdS)Nzexpected to get z0, Tensor-like class, or None as an argument got z insteadz Use push_ú_mode instead.z¼If you intended to completely override the preexisting mode, pass ignore_preexisting=True. This can result in unexpected behavior; please consider rewriting your mode to be a subclass of z to make it compositional!zAttempted to enable_z+_mode, but there is already an active mode z. Ú__zThe argument passed to enable_z_mode must implement ) Ú isinstancerr&Ú issubclassÚ ValueErrorrr#rÚhasattrr%)r$r)r'r(ÚoldÚ help_textZ required_fnr r r Ú _enable_modeAs<ÿ þý ý ÿÿ ÿ  r3c cs°t||jƒr8td| ¡›d| ¡›d|›d|j›d ƒ‚| ¡}|dkrV|jdd}n|}||d}t||jƒsŠtd|j›d| ¡›ƒ‚| |¡z |VW5| |¡XdS) Nz Expected a z. constructor function, but got an instance of ú z. Consider using enable_r+rzThe callable passed to push_z_modemust return a )r-rr/rrr#rr%)Zctorr)r1rr$r r r Ú _push_models" &ÿ  ÿ  r5) rÚtypingrZ dataclassesrrrrr3rr5r r r r Ús  +