Sleep 1 s

apache · Jul 1, 2024 · f41cb66 · f41cb66
1 parent ae2d125
commit f41cb66
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/gluten-core/src/main/java/org/apache/gluten/vectorized/JniLibLoader.java b/gluten-core/src/main/java/org/apache/gluten/vectorized/JniLibLoader.java
@@ -19,6 +19,7 @@
 import org.apache.gluten.exception.GlutenException;
 
 import org.apache.spark.util.GlutenShutdownManager;
+import org.apache.spark.util.Utils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -77,6 +78,13 @@ public class JniLibLoader {
   }
 
   public static synchronized void forceUnloadAll() {
+    if (Utils.isTesting()) {
+      try {
+        Thread.sleep(1000);
+      } catch (InterruptedException e) {
+        throw new RuntimeException(e);
+      }
+    }
     List<String> loaded = new ArrayList<>(REQUIRE_UNLOAD_LIBRARY_PATHS);
     Collections.reverse(loaded); // use reversed order to unload
     loaded.forEach(JniLibLoader::unloadFromPath);