revert exception

apache · Dec 23, 2024 · f8cd555 · f8cd555
1 parent 5bcf9fc
commit f8cd555
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...1.20/src/main/java/org/apache/celeborn/plugin/flink/tiered/CelebornTierConsumerAgent.java b/...1.20/src/main/java/org/apache/celeborn/plugin/flink/tiered/CelebornTierConsumerAgent.java
@@ -57,7 +57,7 @@
 import org.slf4j.LoggerFactory;
 
 import org.apache.celeborn.common.CelebornConf;
-import org.apache.celeborn.common.exception.CelebornIOException;
+import org.apache.celeborn.common.exception.DriverChangedException;
 import org.apache.celeborn.common.exception.PartitionUnRetryAbleException;
 import org.apache.celeborn.common.identity.UserIdentifier;
 import org.apache.celeborn.plugin.flink.RemoteShuffleResource;
@@ -332,7 +332,7 @@ private void initShuffleClient(TierShuffleDescriptorImpl remoteShuffleDescriptor
               conf,
               new UserIdentifier("default", "default"),
               bufferSizeBytes);
-    } catch (CelebornIOException e) {
+    } catch (DriverChangedException e) {
       throw new RuntimeException(e.getMessage());
     }
   }