&`i% dZddlmZmZddlmZmZmZmZm Z m Z ddl m Z ddl mZerddlmZGddeZe d e ZGd d eeeZGd dee eefZdS)z>Ranker component for operator selection in streaming executor.)ABCabstractmethod) TYPE_CHECKINGGenericListProtocolTupleTypeVar)PhysicalOperator)ResourceManager)TopologycbeZdZdZdddefdZdddefdZdddefdZdddefdZdddefdZ d S) Comparablez4Protocol for types that can be compared for ranking.otherreturncdSNselfrs w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/data/_internal/execution/ranker.py__lt__zComparable.__lt__ cdSrrrs r__le__zComparable.__le__rrcdSrrrs r__gt__zComparable.__gt__rrcdSrrrs r__ge__zComparable.__ge__rrcdSrrrs r__eq__zComparable.__eq__rrN) __name__ __module__ __qualname____doc__boolrrrr r"rrrrr s>> L T     L T     L T     L T     L T      rr RankingValue)boundc jeZdZdZededddedefdZde edddede efd Z d S) Rankerz4Abstract base class for operator ranking strategies.optopologyr resource_managerrcdS)aRank operator for execution priority. Operator to run next is selected as the one with the *smallest* value of the lexicographically ordered ranks composed of (in order): Args: op: Operator to rank topology: Current execution topology resource_manager: Resource manager for usage information Returns: Rank (tuple) for operator Nr)rr,r-r.s r rank_operatorzRanker.rank_operator's ( ropscRt|dksJfd|DS)Nrc>g|]}|Sr)r0).0r,r.rr-s r z)Ranker.rank_operators..Es,QQQr""2x1ABBQQQr)len)rr1r-r.s` ``rrank_operatorszRanker.rank_operators=s:3xx!||||QQQQQQSQQQQrN) r#r$r%r&rr r r(r0rr7rrrr+r+$s>>   *      ^ *R " #RR* R l  RRRRRRrr+c :eZdZdZdedddedeeeffdZdS) DefaultRankerzRanker implementation.r,r-r r.rcj|rdnd}|||jfS)aComputes rank for op. *Lower means better rank* 1. Whether operator's could be throttled (int) 2. Operators' object store utilization Args: op: Operator to rank topology: Current execution topology resource_manager: Resource manager for usage information Returns: Rank (tuple) for operator r)throttling_disabled get_op_usageobject_store_memory)rr,r-r.r<s rr0zDefaultRanker.rank_operatorKsB($ #9#;#;Baa  ) )" - - A  rN) r#r$r%r&r r r intr0rrrr9r9Hs\     *  sCx       rr9N)r&abcrrtypingrrrrr r 'ray.data._internal.execution.interfacesr -ray.data._internal.execution.resource_managerr 5ray.data._internal.execution.streaming_executor_stater rr(r+r?r9rrrrEsMDD########IIIIIIIIIIIIIIIIDDDDDDIIIIIIONNNNNN        (w~Z888 !R!R!R!R!RS','!R!R!RH     F5c?+     r