o $iR @sTddlmZmZmZmZddlmZddlmZddl m Z e GdddeZ dS))AnyDictListUnion) StatsBase)single_value_to_cpu) DeveloperAPIc seZdZdZdZfddZdeeefffdd Z deeefdd ffd d Z de fd d Z dde deedffddZdedd fddZdeddd fddZ dde de deeeeffddZdefddZZS) 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. itemcstj|i|d|_dS)z!Initializes a ItemStats instance.N)super__init___item)selfargskwargs __class___/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/rllib/utils/metrics/stats/item.pyr s zItemStats.__init__returncst}|j|d<|SNr )r get_stater rstaterrrrs  zItemStats.get_staterNcst||d|_dSr)r set_stater rrrrr"s zItemStats.set_statecCsdS)Nrrrrr__len__&szItemStats.__len__TcompilecCs.|j}d|_t|}|r|S|}||_|S)N)r rclone)rrr return_statsrrrreduce)szItemStats.reducecCst||_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 rrrpush6s zItemStats.pushincoming_statscCs$t|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. rz_ItemStats should only be merged with one other ItemStats object which replaces the current itemrN)lenr )rr$rrrmergeCs zItemStats.mergeFlatest_merged_onlycCst|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")rrr'r rrrpeekRs zItemStats.peekcCsd|S)Nz ItemStats()r(rrrr__repr__kszItemStats.__repr__)T)TF)__name__ __module__ __qualname____doc__stats_cls_identifierr rstrrrrintrboolrr!r#rr&r(r) __classcell__rrrrr s&    r N) typingrrrr"ray.rllib.utils.metrics.stats.baser#ray.rllib.utils.metrics.stats.utilsrray.util.annotationsrr rrrrs