&`iXDddlZddlmZddlmZmZGddZdS)N)deque)DequeTuplec6eZdZdZdefdZdefdZdZdZdS) TimeWindowAverageCalculatorzMA utility class to calculate the average of values reported in a time window.window_scX|dksJ||_t|_d|_dS)Nr) _window_sr_values_sum)selfrs y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/data/_internal/average_calculator.py__init__z$TimeWindowAverageCalculator.__init__ s0!||||!4977  valuectj}|j||f|xj|z c_||dS)z!Report a value to the calculator.N)timer appendr _trim)r rnows rreportz"TimeWindowAverageCalculator.reportsJikk S%L))) U  3rc|tjt|jdkrdS|jt|jz S)z~Get the average of values reported in the time window, or None if no values reported in the last time window. rN)rrlenr r )r s r get_averagez'TimeWindowAverageCalculator.get_averagesM 49;; t|   ! !4y3t|,,,,rcFt|jdkr||jddz |jkri|j\}}|xj|zc_t|jdkr#||jddz |jkcdSdSdSdS)z6Remove the values reported outside of the time window.rN)rr r popleftr )r r_rs rrz!TimeWindowAverageCalculator._trim&s$,!##dl1oa.@(@4>(Q(Q|++--HAu II II$,!##dl1oa.@(@4>(Q(Q(Q(Q####(Q(QrN) __name__ __module__ __qualname____doc__floatrrrrrrrrsoWW     E---rr)r collectionsrtypingrrrr#rrr&sj $$$$$$$$$$r