&`i dddlmZmZddlmZddlmZddddddZd Zd e d e d ee fd Z dS))DecimalInvalidOperation)reduce)Optionalv4v5ev5pv6e)ztpu-v4-podsliceztpu-v5-lite-deviceztpu-v5-lite-podslicez tpu-v5p-slicez tpu-v6e-slicec t|tttfrt|Sdddddddddd d }t |}|}d }t |dkr+|dd kr|d|vr|d d}|dd }n1t |dkr|d|vr|d d}|dd } t|}n0#t $r#td |wxYw||S| d rd}n8t |dkrd}n"td||dkr"td||d|vr"td|t||d}|||zzS)aParse kubernetes canonical form quantity like 200Mi to a decimal number. Supported SI suffixes: base1024: Ki | Mi | Gi | Ti | Pi | Ei base1000: n | u | m | "" | k | M | G | T | P | E See https://github.com/kubernetes/apimachinery/blob/master/pkg/api/resource/quantity.go Args: quantity: string. kubernetes canonical form quantity Returns: Decimal: The parsed quantity as a decimal number Raises: ValueError: On invalid or unknown input ) numKkMGTPENizInvalid number format: {}iiz{} has unknown suffixkir) isinstanceintfloatrstrlenr ValueErrorformatendswith)quantity exponentsnumbersuffixbaseexponents y/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/autoscaler/_private/kuberay/utils.pyparse_quantityr0s$(S%122!x              I8}}H F F 8}}hrlc11 B<9 $ $crc]Fbcc]F X!   9 9#2#"##E EEE4;;FCCDDDE~  sC V  077AABBB~~077AABBB ay !!077AABBBy+,,H T8^ $$s C-C?topology acceleratorreturnc|r[|rYt|}d|dD}td|}d}|dks|dkrd}||z}|d|Sd S) aConvert Kubernetes gke-tpu nodeSelectors to TPU accelerator_type for a kuberay TPU worker group. Args: topology: value of the cloud.google.com/gke-tpu-topology Kubernetes nodeSelector, describes the physical topology of the TPU podslice. accelerator: value of the cloud.google.com/gke-tpu-accelerator nodeSelector, the name of the TPU accelerator, e.g. tpu-v4-podslice Returns: A string, accelerator_type, e.g. "v4-8". c,g|]}t|S)r").0 chip_counts r/ z.tpu_node_selectors_to_type..gsQQQz3z??QQQxc ||zS)Nr6)r;ys r/z,tpu_node_selectors_to_type..hs Ar:rrr r-N)!gke_tpu_accelerator_to_generationsplitr)r1r2 generationchip_dimensions num_chipsdefault_num_cores_per_chip num_coress r/tpu_node_selectors_to_typerGYs +K +6{C QQX^^C=P=PQQQ--?? %&"   u!4!4)* & :: **y*** 4r:N) decimalrr functoolsrtypingrr@r0r$rGr6r:r/rKs.------- ! %%!D%D%D%N38C=r: