o ci"@sddlmZmZddlmZddlmZmZmZm Z m Z ddl m Z dZ dZe dZe ded ZGd d d eZGd d d eZGdddeeeZdS))ABCabstractmethod)Enum)DictGenericOptionalTupleTypeVar)Gaugeray_train_run_nameray_train_run_idTE)boundc @steZdZdedededeeeffddZedefdd Zed efd d Z ed dZ d e edffddZ dS)Metricnamedefault description base_tagscCs$||_||_t|||d|_dS)a Initialize a new metric. Args: name: The name of the metric. default: The default value of the metric. description: The description of the metric. base_tags: The base tags for the metric. )rtag_keysN)_default _base_tagsr _get_tag_keys_gauge)selfrrrrrW/home/ubuntu/.local/lib/python3.10/site-packages/ray/train/v2/_internal/metrics/base.py__init__s zMetric.__init__valuecCdS)zhUpdate the metric value. Args: value: The value to update the metric with. Nrrrrrrrecord'sz Metric.recordreturncCr)zGet the value of the metric. Returns: The value of the metric. If the metric has not been recorded, the default value is returned. Nrrrrr get_value0szMetric.get_valuecCr)z$Reset values and clean up resources.Nrr#rrrreset:sz Metric.reset.cCst|jSN)tuplerkeysr#rrrr?szMetric._get_tag_keysN) __name__ __module__ __qualname__strr rrrr!r$r%rrrrrrrs"   rcsZeZdZdZdededeeefffdd Zdefdd Zd efd d Z d dZ Z S) TimeMetricz#A metric for tracking elapsed time.rrrcsd|_tj|d||ddS)Ngrrrr)_current_valuesuperr)rrrr __class__rrrFs zTimeMetric.__init__rcCs$|j|7_|j|j|jdS)zUpdate the time metric value by accumulating the time. Args: value: The time value to increment the metric by. N)r/rsetrr rrrr!TszTimeMetric.recordr"cCs|jSr&)r/r#rrrr$]szTimeMetric.get_valuecCs|j|_|j|j|jdSr&)rr/rr3rr#rrrr%`szTimeMetric.reset) r)r*r+__doc__r,rrfloatr!r$r% __classcell__rrr1rr-Cs  r-cseZdZdZdZdZdededeeefdeffdd Zd e fd d Z d e d e fddZ ddZ d eedffddZd e d eeeffddZZS) EnumMetricz"A metric for tracking enum values.rrrr enum_tag_keycs&||_d|_tj||j||ddS)Nr.) _enum_tag_keyr/r0r DEFAULT_VALUE)rrrrr9r1rrrks zEnumMetric.__init__ enum_valuecCsX||jkrdS|jdur||j}|j|j|||}|j|j|||_dS)zRecord a specific enum value. The metric will be reset to 0 for the previous value and set to 1 for the new value. Args: enum_value: The enum value to record for. N)r/ _get_tagsrr3rRECORDED_VALUE)rr< previous_tags current_tagsrrrr!{s     zEnumMetric.recordr"cCst||jkS)zGet the value for a specific enum value. Args: enum_value: The enum value to get the value for Returns: The value for the enum value )intr/)rr<rrrr$s zEnumMetric.get_valuecCs0|jdur||j}|j|j|d|_dSr&)r/r=rr3r)rtagsrrrr%s   zEnumMetric.reset.cCst|j|jfSr&)r'rr(r:r#rrrrszEnumMetric._get_tag_keyscCs|j}|j||j<|Sr&)rcopyrr:)rr<rBrrrr=s  zEnumMetric._get_tags)r)r*r+r4r;r>r,rrrr!rAr$r%rrr=r6rrr1rr7es$  "r7N)abcrrenumrtypingrrrrr ray.util.metricsr RUN_NAME_TAG_KEYRUN_ID_TAG_KEYr rrr-r7rrrrs   5"