Merge pull request jax-ml#25128 from ROCm:ci_fix_wheelhouse_relative_paths-upstream

PiperOrigin-RevId: 701143534

Author: Google-ML-Automation

build/rocm/ci_build

@@ -89,9 +89,9 @@ def dist_wheels(
 
     mounts = [
         "-v",
-        "./:/jax",
+        os.path.abspath("./") + ":/jax",
         "-v",
-        "./wheelhouse:/wheelhouse",
+        os.path.abspath("./wheelhouse") + ":/wheelhouse",
     ]
 
     if xla_path:
@@ -210,7 +210,7 @@ def test(image_name):
     # JAX and jaxlib are already installed from wheels
     mounts = [
         "-v",
-        "./:/jax",
+        os.path.abspath("./") + ":/jax",
     ]
 
     cmd.extend(mounts)