&`ixddlmZmZddlmZmZmZmZmZedZ edZ GddZ dS))Counter defaultdict)Dict GeneratorListOptionalTypeVarTUceZdZdZddefdZedZedZdde d e d d fd Z dde d e d d fd Z de d efdZ de ded efdZde d eefdZdded eed d ffdZd S) _ObjectCacheaCache up to some maximum count given a grouping key. This object cache can e.g. be used to cache Ray Tune trainable actors given their resource requirements (reuse_actors=True). If the max number of cached objects for a grouping key is reached, no more objects for this group will be cached. However, if `may_keep_one=True`, one object (globally across all grouping keys) may be cached, even if the max number of objects is 0. This is to allow to cache an object if the max number of objects of this key will increase shortly after (as is the case e.g. in the Ray Tune control loop). Args: may_keep_one: If True, one object (globally) may be cached if no desired maximum objects are defined. T may_keep_oneczd|_tt|_t |_||_dS)Nr)_num_cached_objectsrlist_cached_objectsr_max_num_objects _may_keep_one)selfrs o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/tune/utils/object_cache.py__init__z_ObjectCache.__init__s4() 1z,_ObjectCache.cache_object..js"==1======rr!T)lenrrrranyrappend)rr"r0s r cache_objectz_ObjectCache.cache_objectLs( t#C( ) )T-B3-G G G% u'!++u==d3::<<===== u S!((---   A%  trc||sdS|xjdzc_|j|dS)zGet one cached object for a key. This will remove the object from the cache. Args: key: Group key. Returns: Cached object. Nr!r)r/rrpopr.s rpop_cached_objectz_ObjectCache.pop_cached_objectssO%%c** 4   A%  #C(,,Q///rF force_allc#K|jo| }|jD]\}}|s |j|nd}|jdkr5|r3t d|jDsdSt||kr:|xjdzc_|dVt||k:dS)aReturn a generator over cached objects evicted from the cache. This method yields all cached objects that should be evicted from the cache for cleanup by the caller. If the number of max objects is lower than the number of cached objects for a given key, objects are evicted until the numbers are equal. If `max_keep_one=True` (and ``force_all=False``), one cached object may be retained. Objects are evicted FIFO. If ``force_all=True``, all objects are evicted. Args: force_all: If True, all objects are flushed. This takes precedence over ``keep_one``. Yields: Evicted objects to be cleaned up by caller. rr!c3K|]}|VdSrr3r4s rr7z4_ObjectCache.flush_cached_objects..s"FF!AFFFFFFrN) rritemsrrr9rr8r=)rr?keep_oner"objs max_cacheds rflush_cached_objectsz!_ObjectCache.flush_cached_objectss4%7i--3355 " "IC;DK.s33!J(A--.FFt'<'C'C'E'EFFFFF. d))j((((A-((hhqkk!!!d))j(( " "r)T)r!)F)__name__ __module__ __qualname____doc__r-rpropertyrr r intr(r+r/r r;rr>rrFr3rrr r su(**T****((X(33X3)))s)4)))))))s)4)))) /Q /4 / / / /%%%d%%%%N0Q08A;0000")")"d)"yDRV?W)")")")")")"rr N) collectionsrrtypingrrrrr r r r r3rrrOs,,,,,,,,;;;;;;;;;;;;;; GCLL GCLLc"c"c"c"c"c"c"c"c"c"r