ukiUddlmZddlZddlmZddlZdZdededdfdZdedede fd Z eGd d Z iZ e ee fed <d eddfdZd eddfdZd eddfdZd ede fdZd edededdfdZedededededededededededed ed!y)") dataclassN) ModuleTypec<tjfd}|S)Nc|vr3|\}}| t|tj|td|Stdd|)N) stacklevelzmodule z has no attribute )AttributeErrorwarningswarnDeprecationWarning)namemessagefn deprecationsmodules P/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/deprecations.pygetattrz$deprecation_getattr..getattr1sX | &kgr W%%mmG/A> i 76*,>thG HH) functoolscache)rrrs`` rdeprecation_getattrr0s" ??II .rrr returncJ|j|\}}|df|j|<y)zAccelerate the deprecation of a module-level attribute. Raises an AttributeError instead of a DeprecationWarning upon attribute access. Used in Google-internal code to implement faster deprecation. N _deprecations)rr r_s raccelerate_getattr_deprecationr>s, ##D)*'1 '&trc*|j|dduS)zqReturns true if given name is accelerated. Raises an error if name is not a deprecated attribute in module. Nr)rr s ris_accelerated_attributer Gs   d #A &$ ..rceZdZUdZeed<y)DeprecationStateF acceleratedN)__name__ __module__ __qualname__r#bool__annotations__rrr"r"Vs +trr"_registered_deprecationsdeprecation_idc&tt|<y)N)r"r*r+s rregisterr.]s-=-?>*rc\|tvrtd|dtj|yNdeprecation_id= not registered.)r* ValueErrorpopr-s r unregisterr5as/33 **:; <<~.rcN|tvrtd|ddt|_y)Nr1r2Tr*r3r#r-s r accelerater8gs/33 **:; <<9=>*6rcV|tvrtd|dt|jSr0r7r-s ris_acceleratedr:ms233 **:; << !. 1 = ==rrrcnt|r t|tj|t|dzy)zGWarns about a deprecation, or errors if the deprecation is accelerated.r)categoryrN)r:r3r r r )r+rrs rr r ss-N# W  MM'$6'!^-rzdefault-dtype-bits-configzjax-checkpoint-concretezjax-lax-dot-positional-argszjax-lib-modulezjax-nn-one-hot-float-inputzjax-numpy-arange-complexz jax-numpy-astype-complex-to-realzjax-numpy-clip-argszjax-scipy-special-sph-harmzsafer-randint-configzjax-pmap-no-rank-reductionzjax-make-mesh-default-explicitzpltpu-memory-space-any) dataclassesrrtypesrr rstrrr'r r"r*dictr(r.r5r8r:intr r)rrrBsd"< /:/S/T//Z/s/t/  9;$s$445:@S@T@/s/t/ >s>t> >3>4> --s--- $% "# &'  %& #$ +,  %&   %& )* !"r