o ¡…›iIã@s„ddlZddlmZddlmZmZddlmZedƒZedƒZ edƒZ e ƒZ deee efe fd eee efe ffd d „Z dS) éN)ÚCallable)Ú ConcatenateÚTypeVar)Ú ParamSpecÚ_PÚ_TÚ_CÚfÚreturnc s<dˆj‰t ˆ¡dtdtjdtjdtf‡‡fdd„ ƒ}|S)a Like `@functools.cache` but for methods. `@functools.cache` (and similarly `@functools.lru_cache`) shouldn't be used on methods because it caches `self`, keeping it alive forever. `@cache_method` ignores `self` so won't keep `self` alive (assuming no cycles with `self` in the parameters). Footgun warning: This decorator completely ignores self's properties so only use it when you know that self is frozen or won't change in a meaningful way (such as the wrapped function being pure). Ú_cache_method_ÚselfÚargsÚkwargsr csf|rtdƒ‚t|ˆdƒ}si}t|ˆ|ƒ| |t¡}|tur"|Sˆ|g|¢Ri|¤Ž}|||<|S)Nz.cache_method does not accept keyword arguments)ÚAssertionErrorÚgetattrÚsetattrÚgetÚ_cache_sentinel)r r rÚcacheÚ cached_valueÚvalue©Ú cache_namer ©úX/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/torch/utils/_functools.pyÚwrap s  zcache_method..wrap)Ú__name__Ú functoolsÚwrapsrrr rr)r rrrrÚ cache_methods &r)rÚcollections.abcrÚtypingrrÚtyping_extensionsrrrrÚobjectrrrrrrÚs  ÿþ