o 2wi @sUdZddlZddlmZmZmZddlmZddlm Z ddZ eded efd Z Gd d d e Z e ejd dejddejddeejddeejdddZe ed<dS)zFunction cachingN)AnyCallableTypeVar)Memory) FunctionMakercCstj|dt||d|dS)Nz"return decfunc(%(shortsignature)s))decfunc) __wrapped__)rcreatedict)decfuncr K/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/librosa/_cache.py_decorator_apply s  r_F.)boundc@seZdZdZdedefddZdedeegeffdd Z dededd fd d Z dededefd dZ dededefddZ dededd fddZ dededd fddZd S) CacheManagera)The librosa cache manager class wraps joblib.Memory with a __call__ attribute, so that it may act as a function. Additionally, it provides a caching level filter, so that different functions can be cached or not depending on the user's preference for speed vs. storage usage. argskwargscOs&|dd}t|i||_||_dS)Nlevel )poprmemoryr)selfrrrr r r__init__ s  zCacheManager.__init__rreturncsfdd}|S)z Cache with an explicitly defined level. Example usage: @cache(level=2) def semi_important_function(some_arguments): ... cs(jjdurjkrtjj|S|S)z4Add an input/output cache to the specified function.N)rlocationrrcache)functionrrr rwrapper4sz&CacheManager.__call__..wrapperr )rrr r rr__call__)s zCacheManager.__call__NcO|jj|i|dS)zClear the cacheN)rclearrrrr r rr#>zCacheManager.clearcO|jj|i|S)zEvaluate a function)revalr$r r rr'BzCacheManager.evalcOr&)z0Return the formatted representation of an object)rformatr$r r rr)Fr(zCacheManager.formatcOr")zReduce the size of the cacheN)r reduce_sizer$r r rr*Jr%zCacheManager.reduce_sizecOr")zRaise a warningN)rwarnr$r r rr+Nr%zCacheManager.warn)__name__ __module__ __qualname____doc__rrintrrr!r#r'r)r*r+r r r rrs rLIBROSA_CACHE_DIRLIBROSA_CACHE_MMAPLIBROSA_CACHE_COMPRESSFLIBROSA_CACHE_VERBOSELIBROSA_CACHE_LEVELr) mmap_modecompressverboserr)r/ostypingrrrjoblibr decoratorrrrobjectrenvirongetr0r__annotations__r r r rs   =