&`iR nddlmZmZmZmZddlmZddlmZddl m Z e GddeZ dS))AnyDictListUnion) StatsBase)single_value_to_cpu) DeveloperAPIc eZdZdZdZfdZdeeefffd Z deeefddffd Z de fd Z dd e deedffd Zdeddfd ZdedddfdZ dd e de deeeeffdZdefdZxZS) ItemStatsaA Stats object that tracks a single item. Note the follwing limitation: That, when calling `ItemStats.merge()`, we replace the current item. This is because there can only be a single item tracked by definition. This class will check if the logged item is a GPU tensor. If it is, it will be converted to CPU memory. Use this if you want to track a single item that should not be reduced. An example would be to log the total loss. itemcHtj|i|d|_dS)z!Initializes a ItemStats instance.N)super__init___item)selfargskwargs __class__s v/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/rllib/utils/metrics/stats/item.pyrzItemStats.__init__s*$)&))) returnc\t}|j|d<|SNr )r get_staterrstaters rrzItemStats.get_states(!!## f  rrNcdt||d|_dSr)r set_staterrs rrzItemStats.set_state"s* %   6] rcdS)Nrs r__len__zItemStats.__len__&sqrTcompilec~|j}d|_t|}|r|S|}||_|S)N)rrclone)rr$r return_statss rreducezItemStats.reduce)sEz "4((  Kzz|| ! rc.t||_dS)zPushes a value into this Stats object. Args: item: The value to push. Can be of any type. GPU tensors are moved to CPU memory. Returns: None Nrr)rr s rpushzItemStats.push6s).. rincoming_statscdt|dks Jd|dj|_dS)zMerges ItemStats objects. Args: incoming_stats: The list of ItemStats objects to merge. Returns: None. The merge operation modifies self in place. r z_ItemStats should only be merged with one other ItemStats object which replaces the current itemrN)lenr)rr,s rmergezItemStats.mergeCs=   1 $ $ $ l % $ $$A&, rFlatest_merged_onlyc8t|j}|r|S|gS)a Returns the internal item. This does not alter the internal item. Args: compile: If True, return the internal item directly. If False, return the internal item as a single-element list. latest_merged_only: This parameter is ignored for ItemStats. ItemStats tracks a single item, not a series of merged values. The current item is always returned regardless of this parameter. Returns: The internal item. r*)rr$r0r s rpeekzItemStats.peekRs'(#4:..  Kv rc0d|S)Nz ItemStats()r2r"s r__repr__zItemStats.__repr__ks)DIIKK)))r)T)TF)__name__ __module__ __qualname____doc__stats_cls_identifierrrstrrrrintr#boolrr(r+rr/r2r4 __classcell__)rs@rr r s  " 4S> #tCH~#$######  d eC4D.E     / / / / / / -D$5 -$ - - - - @E8< sDI~ 2*#********rr N) typingrrrr"ray.rllib.utils.metrics.stats.baser#ray.rllib.utils.metrics.stats.utilsrray.util.annotationsr r r!rrrBs))))))))))))888888CCCCCC------c*c*c*c*c* c*c*c*c*c*r