diff --git a/sql/core/src/main/scala/org/apache/spark/sql/SparkSessionExtensions.scala b/sql/core/src/main/scala/org/apache/spark/sql/SparkSessionExtensions.scala index 6b3744fe02d2..dc3dfaae4ff7 100644 --- a/sql/core/src/main/scala/org/apache/spark/sql/SparkSessionExtensions.scala +++ b/sql/core/src/main/scala/org/apache/spark/sql/SparkSessionExtensions.scala @@ -230,7 +230,7 @@ class SparkSessionExtensions { * cache hit rate by normalizing different plans to the same form. These rules should never modify * the result of the LogicalPlan. */ - def injectPlanNormalizationRules(builder: RuleBuilder): Unit = { + def injectPlanNormalizationRule(builder: RuleBuilder): Unit = { planNormalizationRules += builder } diff --git a/sql/core/src/test/scala/org/apache/spark/sql/SparkSessionExtensionSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/SparkSessionExtensionSuite.scala index f5f04eabec03..66507756e892 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/SparkSessionExtensionSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/SparkSessionExtensionSuite.scala @@ -194,7 +194,7 @@ class SparkSessionExtensionSuite extends SparkFunSuite with SQLHelper { test("inject plan normalization rules") { val extensions = create { extensions => - extensions.injectPlanNormalizationRules { session => + extensions.injectPlanNormalizationRule { session => org.apache.spark.sql.catalyst.optimizer.PushDownPredicates } }