fix ut

apache · Feb 18, 2025 · 5132a25 · 5132a25
1 parent d4db85d
commit 5132a25
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.../src/main/java/org/apache/celeborn/service/deploy/worker/storage/PartitionDataWriter.java b/.../src/main/java/org/apache/celeborn/service/deploy/worker/storage/PartitionDataWriter.java
@@ -89,8 +89,9 @@ public PartitionDataWriter(
   }
 
   public DiskFileInfo getDiskFileInfo() {
+    // keep compatible with current logic
     if (!(tierWriterProxy.getCurrentFileInfo() instanceof DiskFileInfo)) {
-      throw new IllegalStateException("tier writer file info needs to be disk file info");
+      return null;
     }
     return ((DiskFileInfo) tierWriterProxy.getCurrentFileInfo());
   }