guard -it for docker with check

ROCm · Oct 2, 2024 · 6a0d1a6 · 6a0d1a6
1 parent 8a6a2c9
commit 6a0d1a6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/build/rocm/ci_build b/build/rocm/ci_build
@@ -99,7 +99,7 @@ def dist_wheels(
 
     cmd.extend(mounts)
 
-    if os.isatty(sys.stdout.fileno()) and os.isatty(sys.stdin.fileno()):
+    if os.isatty(sys.stdout.fileno()):
         cmd.append("-it")
 
     # NOTE(mrodden): bazel times out without --init, probably blocking on a zombie PID
@@ -208,10 +208,12 @@ def test(image_name):
     cmd = [
         "docker",
         "run",
-        "-it",
         "--rm",
     ]
 
+    if os.isatty(sys.stdout.fileno()):
+        cmd.append("-it")
+
     # NOTE(mrodden): we need jax source dir for the unit test code only,
     # JAX and jaxlib are already installed from wheels
     # docker run fails when mounting ./