&`iddlZddlmZddlmZddlmZeGddZeGddZdS) N)List) Deprecated)_Timerc$eZdZdZdZdZdZdS)MetricsContextaMetrics context object for a local iterator. This object is accessible by all operators of a local iterator. It can be used to store and retrieve global execution metrics for the iterator. It can be accessed by calling LocalIterator.get_metrics(), which is only allowable inside iterator functions. Attributes: counters: dict storing increasing metrics. timers: dict storing latency timers. info: dict storing misc metric values. current_actor: reference to the actor handle that produced the current iterator output. This is automatically set for gather_async(). ctjt|_tjt|_i|_d|_dSN) collections defaultdictintcountersrtimersinfo current_actorselfs i/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/iter_metrics.py__init__zMetricsContext.__init__s;#/44 !-f55  !cVt|jt|jddS)z+Return a serializable copy of this context.N)r rr)dictr rrs rsavezMetricsContext.save s.T]++OO   rc|j|j|d|j|d|_dS)z*Restores state given the output of save().r rN)r clearupdaterr)rvaluess rrestorezMetricsContext.restore(sT  VJ/000 6N rN)__name__ __module__ __qualname____doc__rrrrrrrsK """    #####rrc>eZdZdZ ddededfdZdZdZdS) SharedMetricszHolds an indirect reference to a (shared) metrics context. This is used by LocalIterator.union() to point the metrics contexts of entirely separate iterator chains to the same underlying context.Nmetricsparentscv|p t|_|pg|_||jdSr )rr%r&set)rr%r&s rrzSharedMetrics.__init__7s;2."2"2 }"  rcR||_|jD]}||dS)z>Recursively set self and parents to point to the same metrics.N)r%r&r()rr%parents rr(zSharedMetrics.set>s9 l  F JJw      rc|jSr )r%rs rgetzSharedMetrics.getDs |r)NN) rrr r!rrrr(r,r"rrr$r$0ssII PT%7;O7L   rr$) r typingrray.util.annotationsrray.util.timerrrr$r"rrr0s++++++!!!!!! $#$#$#$#$#$#$# $#N  r