o $i@sTddlZddlmZddlmZddlmZeGdddZeGdddZdS) N)List) Deprecated)_Timerc@s(eZdZdZddZddZddZdS) 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(). cCs(tt|_tt|_i|_d|_dSN) collections defaultdictintcountersrtimersinfo current_actorselfrR/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/util/iter_metrics.py__init__s   zMetricsContext.__init__cCst|jt|jddS)z+Return a serializable copy of this context.N)r r r )dictr r rrrrsave szMetricsContext.savecCs2|j|j|d|j|d|_dS)z*Restores state given the output of save().r r N)r clearupdater r )rvaluesrrrrestore(s  zMetricsContext.restoreN)__name__ __module__ __qualname____doc__rrrrrrrrs  rc@s:eZdZdZ d dededfddZddZd d ZdS) 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.NmetricsparentscCs&|pt|_|p g|_||jdSr)rrrset)rrrrrrr7s  zSharedMetrics.__init__cCs ||_|jD]}||qdS)z>Recursively set self and parents to point to the same metrics.N)rrr )rrparentrrrr >s  zSharedMetrics.setcCs|jSr)rrrrrgetDszSharedMetrics.get)NN) rrrrrrrr r"rrrrr0s  r) rtypingrray.util.annotationsrray.util.timerrrrrrrrs   '