tj ddlmZddlZddlZddlmZddlmZmZm Z ddl m Z er ddl m Z ddlmZe dZe d Ze d ZeZed ZddZddZdS)) annotationsN)ThreadPoolExecutor) Concatenate TYPE_CHECKINGTypeVar) ParamSpec)Callable)Future_P_T_C) max_workersfuncCallable[[], _T]returncttjdd fd }d fd }d fd }d fd }||_||_||_|S) z Like functools.cache but with a prefetch() method that starts computing the value in a background thread, and a set() method for prepopulating. Nrr c5turcdddS$dcdddScdddS#1swxYwYdSN)_cache_sentinelresult_cache_future_lockrsT/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/utils/_functools.pywrapperz#prefetchable_cache..wrapper"s   _,,        " ))         TVVF                  s AA AA!AvalNonecr5turtd|ddddS#1swxYwYdS)Nz$prefetchable_cache value already set)r RuntimeError)rrrs rset_valz#prefetchable_cache..set_val.s   _,,"#IJJJF                  s ,00cP5tdddddS#1swxYwYdSr)r)rrrsrclearz!prefetchable_cache..clear5sw   $FG                  s c5tus ddddStddddS#1swxYwYdSr)r_prefetch_executorsubmitrsrprefetchz$prefetchable_cache..prefetch;s  6 6_,,0C 6 6 6 6 6 6 6 6)//55G 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s AAA  A )rr )rr rr)rr)r threadingLocksetr$r()rrr"r$r(rrrs` @@@rprefetchable_cacher,s *F N  E!%G          666666666GKGMG Nf!Callable[Concatenate[_C, _P], _T]c`djztjd fd }|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_selfr args_P.argskwargs _P.kwargsrr c|rtdt|dx}si}t||||t}|tur|S|g|Ri|}|||<|S)Nz.cache_method does not accept keyword arguments)AssertionErrorgetattrsetattrgetr)r2r3r5cache cached_valuevalue cache_namer.s rwrapzcache_method..wrapYs  S !QRR R z4888 -E D*e , , ,yy77  . . $((((((d  r-)r2r r3r4r5r6rr )__name__ functoolswraps)r.r@r?s` @r cache_methodrDHsQ"AJ.J_Q        Kr-)rrrr)r.r/rr/) __future__rrBr)concurrent.futuresrtypingrrrtyping_extensionsrcollections.abcr r r r r objectrr&r,rDr-rrLs""""""1111116666666666''''''*(((((())))))Yt__ WT]] WT]]&((''A666,,,,^r-