diff --git a/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/Transition.scala b/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/Transition.scala index 833592294532..0d9488d827c2 100644 --- a/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/Transition.scala +++ b/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/Transition.scala @@ -99,7 +99,7 @@ object Transition { graphCache.getOrElseUpdate( aliasOrClass, { val base = GlutenCostModel.find(aliasOrClass) - graphBuilder.build(TransitionGraph.newCostModel(base)) + graphBuilder.build(TransitionGraph.asTransitionCostModel(base)) }) } diff --git a/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/TransitionGraph.scala b/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/TransitionGraph.scala index 2a71646f1605..a5b4612cbf16 100644 --- a/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/TransitionGraph.scala +++ b/gluten-core/src/main/scala/org/apache/gluten/extension/columnar/transition/TransitionGraph.scala @@ -54,7 +54,7 @@ object TransitionGraph { FloydWarshallGraph.builder() } - private[transition] def newCostModel( + private[transition] def asTransitionCostModel( base: GlutenCostModel): FloydWarshallGraph.CostModel[Transition] = { new TransitionCostModel(base) }