Update sql-plugin/src/main/scala/org/apache/spark/sql/rapids/GpuShuff…

…leEnv.scala Co-authored-by: Gera Shegalov <[email protected]>
NVIDIA · May 25, 2024 · d129d76 · d129d76
1 parent 3a109d4
commit d129d76
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/GpuShuffleEnv.scala b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/GpuShuffleEnv.scala
@@ -110,9 +110,9 @@ object GpuShuffleEnv extends Logging {
         rapidsShuffleManager.initialize
       case _ =>
         val rsmLoaderViaShuffleManager = shuffleManager.getClass.getSuperclass.getInterfaces
-        .collectFirst {
-          case c if c.getName == classOf[RapidsShuffleManagerLike].getName => c.getClassLoader
-        }
+          .collectFirst {
+            case c if c.getName == classOf[RapidsShuffleManagerLike].getName => c.getClassLoader
+          }
         val rsmLoaderDirect = classOf[RapidsShuffleManagerLike].getClassLoader
 
         throw new IllegalStateException(s"Cannot initialize the RAPIDS Shuffle Manager " +