diff --git a/client-spark/common/src/main/java/org/apache/spark/shuffle/reader/RssShuffleDataIterator.java b/client-spark/common/src/main/java/org/apache/spark/shuffle/reader/RssShuffleDataIterator.java index 775285ca2f..bd8184c5b3 100644 --- a/client-spark/common/src/main/java/org/apache/spark/shuffle/reader/RssShuffleDataIterator.java +++ b/client-spark/common/src/main/java/org/apache/spark/shuffle/reader/RssShuffleDataIterator.java @@ -38,6 +38,7 @@ import org.apache.uniffle.client.api.ShuffleReadClient; import org.apache.uniffle.client.response.CompressedShuffleBlock; import org.apache.uniffle.common.RssShuffleUtils; +import org.apache.uniffle.common.exception.RssException; public class RssShuffleDataIterator extends AbstractIterator> { @@ -54,6 +55,7 @@ public class RssShuffleDataIterator extends AbstractIterator