fixup

apache · Dec 24, 2024 · 5cf681c · 5cf681c
1 parent 9329a0c
commit 5cf681c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backends-velox/src/main/scala/org/apache/spark/broadcast/BroadcastFactoryUnsafeAccess.scala b/backends-velox/src/main/scala/org/apache/spark/broadcast/BroadcastFactoryUnsafeAccess.scala
@@ -45,7 +45,7 @@ object BroadcastFactoryUnsafeAccess {
     }
   }
 
-  def inject(factory: BroadcastFactory): Unit = {
+  def set(factory: BroadcastFactory): Unit = {
     require(SparkEnv.get != null, "No SparkEnv instance founded")
     val bm = SparkEnv.get.broadcastManager
     bm.synchronized {

diff --git a/...s-velox/src/test/scala/org/apache/spark/broadcast/BroadcastFactoryUnsafeAccessSuite.scala b/...s-velox/src/test/scala/org/apache/spark/broadcast/BroadcastFactoryUnsafeAccessSuite.scala
@@ -34,7 +34,7 @@ class BroadcastFactoryUnsafeAccessSuite extends SharedSparkSession {
     BroadcastFactoryUnsafeAccess.get().stop()
     val factory = new DummyBroadcastFactory()
     assert(!factory.initialized)
-    BroadcastFactoryUnsafeAccess.inject(factory)
+    BroadcastFactoryUnsafeAccess.set(factory)
     assert(BroadcastFactoryUnsafeAccess.get() eq factory)
     assert(factory.initialized)
     val text: String = "DUMMY"