o `Û·i³ã@sTddlZddlmZddlmZddlmZeGdd„dƒƒZeGdd„dƒƒZdS) éN)ÚList)Ú Deprecated)Ú_Timerc@s(eZdZdZdd„Zdd„Zdd„ZdS) ÚMetricsContexta†Metrics 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(t t¡|_t t¡|_i|_d|_dS©N)Ú collectionsÚ defaultdictÚintÚcountersrÚtimersÚinfoÚ current_actor©Úself©rúK/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/util/iter_metrics.pyÚ__init__s   zMetricsContext.__init__cCst|jƒt|jƒddœS)z+Return a serializable copy of this context.N)r r r )Údictr r rrrrÚsave sýzMetricsContext.savecCs2|j ¡|j |d¡|j ¡|d|_dS)z*Restores state given the output of save().r r N)r ÚclearÚupdater r )rÚvaluesrrrÚrestore(s  zMetricsContext.restoreN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrrrrrs  rc@s:eZdZdZ d dededfdd„Zdd„Zd d „ZdS) Ú SharedMetricszÌHolds 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.NÚmetricsÚparentscCs&|ptƒ|_|p g|_| |j¡dSr)rrrÚset)rrrrrrr7s  zSharedMetrics.__init__cCs ||_|jD]}| |¡qdS)z>Recursively set self and parents to point to the same metrics.N)rrr )rrÚparentrrrr >s  ÿzSharedMetrics.setcCs|jSr)rrrrrÚgetDszSharedMetrics.get)NN) rrrrrrrr r"rrrrr0sÿÿ ÿ r) rÚtypingrÚray.util.annotationsrÚray.util.timerrrrrrrrÚs   '