o b²“ijã@s<ddlmZmZddlmZGdd„dƒZGdd„dƒZdS) é)ÚListÚTypeé)ÚPlanc@sVeZdZdZdedefdd„Zedeedfdd„ƒZ edeedfdd „ƒZ d S) ÚRulez%Abstract class for optimization rule.ÚplanÚreturncCót‚)z2Apply the optimization rule to the execution plan.©ÚNotImplementedError)Úselfr©r úc/home/ubuntu/.local/lib/python3.10/site-packages/ray/data/_internal/logical/interfaces/optimizer.pyÚapply sz Rule.applycCógS)z4List of rules that must be applied before this rule.r ©Úclsr r rÚ dependencies ózRule.dependenciescCr)z3List of rules that must be applied after this rule.r rr r rÚ dependentsrzRule.dependentsN) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚ classmethodrrrrr r r rrsrc@s8eZdZdZedeefdd„ƒZdedefdd„Z dS) Ú OptimizerzuAbstract class for optimizers. An optimizers transforms a DAG of operators with a list of predefined rules. rcCr )z,List of predefined rules for this optimizer.r )r r r rÚrulesrzOptimizer.rulesrcCs8|} |jD]}| |¡}q|jj|jjkr |S|}q)z(Optimize operators with a list of rules.)rrÚdagÚdag_str)r rÚ previous_planÚruler r rÚoptimize#s  ÿùzOptimizer.optimizeN) rrrrÚpropertyrrrrr!r r r rrs rN)Útypingrrrrrrr r r rÚs