From d129d76087a81b4db1ba00c78e69b4ea3838976f Mon Sep 17 00:00:00 2001
From: Zach Puller <zach.puller@gmail.com>
Date: Sat, 25 May 2024 11:13:26 -0700
Subject: [PATCH] Update
 sql-plugin/src/main/scala/org/apache/spark/sql/rapids/GpuShuffleEnv.scala

Co-authored-by: Gera Shegalov <gshegalov@nvidia.com>
---
 .../scala/org/apache/spark/sql/rapids/GpuShuffleEnv.scala   | 6 +++---
 1 file changed, 3 insertions(+), 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
index 29959760b8a..cdd8ba5e51f 100644
--- 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 " +