From b6d47928408504a1776ef92fccbda4d3a73ae5a5 Mon Sep 17 00:00:00 2001 From: Balaji Veeramani Date: Mon, 12 Aug 2024 18:22:30 -0500 Subject: [PATCH] [Data] Fix signature of `Rule.plan` (#47094) The method signature is missing self --- python/ray/data/_internal/logical/interfaces/optimizer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/data/_internal/logical/interfaces/optimizer.py b/python/ray/data/_internal/logical/interfaces/optimizer.py index 70b235b42241..6a25a44afe62 100644 --- a/python/ray/data/_internal/logical/interfaces/optimizer.py +++ b/python/ray/data/_internal/logical/interfaces/optimizer.py @@ -6,7 +6,7 @@ class Rule: """Abstract class for optimization rule.""" - def apply(plan: Plan) -> Plan: + def apply(self, plan: Plan) -> Plan: """Apply the optimization rule to the execution plan.""" raise NotImplementedError