fix spark3.1 compile error (#598)

Co-authored-by: zhangli20 <[email protected]>
kwai · Sep 27, 2024 · bde15c2 · bde15c2
1 parent 6918f75
commit bde15c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spark-extension-shims-spark3/src/main/scala/org/apache/spark/sql/blaze/ShimsImpl.scala b/spark-extension-shims-spark3/src/main/scala/org/apache/spark/sql/blaze/ShimsImpl.scala
@@ -698,7 +698,7 @@ class ShimsImpl extends Shims with Logging {
             JniBridge.resourcesMap.put(
               jniResourceId,
               () => {
-                reader.asInstanceOf[BlazeBlockStoreShuffleReader[_, _]].readIpc()
+                reader.asInstanceOf[BlazeBlockStoreShuffleReaderBase[_, _]].readIpc()
               })
 
             pb.PhysicalPlanNode