From 240e6efe210b23261145392888b0c9cc0c1c14cc Mon Sep 17 00:00:00 2001
From: Qing Wang <kingchin1218@gmail.com>
Date: Wed, 5 Jan 2022 09:01:34 +0800
Subject: [PATCH] [Java] Try to fix flaky NamespaceTest (#21370)

---
 java/test/src/main/java/io/ray/test/NamespaceTest.java | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/java/test/src/main/java/io/ray/test/NamespaceTest.java b/java/test/src/main/java/io/ray/test/NamespaceTest.java
index 2248b7bd4..02e2c8234 100644
--- a/java/test/src/main/java/io/ray/test/NamespaceTest.java
+++ b/java/test/src/main/java/io/ray/test/NamespaceTest.java
@@ -113,12 +113,9 @@ public class NamespaceTest {
       builder.redirectError(ProcessBuilder.Redirect.INHERIT);
       driver = builder.start();
       // Wait for driver to start.
-      TimeUnit.SECONDS.sleep(3);
+      driver.waitFor(10, TimeUnit.SECONDS);
       runnable.run();
     } finally {
-      if (driver != null) {
-        driver.waitFor(1, TimeUnit.SECONDS);
-      }
       Ray.shutdown();
     }
   }