fix: Prevent GHA failure when .venv is missing in run_tests_against_package

Author: darisoy

.github/workflows/run_tests_against_package.yml

@@ -162,13 +162,15 @@ jobs:
             # Dynamically discover the 'nvidia' folder and prepend all its sub-library
             # directories (including nccl, cublas, cudnn) to LD_LIBRARY_PATH to prevent
             # JAX from partially loading incompatible system-level CUDA libraries.
-            NVIDIA_DIR=$(find .venv/lib/ -maxdepth 3 -name "nvidia" -type d 2>/dev/null | head -n 1)
-            if [ -n "${NVIDIA_DIR}" ]; then
-              for dir in "${NVIDIA_DIR}"/*; do
-                if [ -d "$dir/lib" ]; then
-                  export LD_LIBRARY_PATH=$(pwd)/$dir/lib:${LD_LIBRARY_PATH}
-                fi
-              done
+            if [ -d ".venv/lib" ]; then
+              NVIDIA_DIR=$(find .venv/lib/ -maxdepth 3 -name "nvidia" -type d 2>/dev/null | head -n 1)
+              if [ -n "${NVIDIA_DIR}" ]; then
+                for dir in "${NVIDIA_DIR}"/*; do
+                  if [ -d "$dir/lib" ]; then
+                    export LD_LIBRARY_PATH=$(pwd)/$dir/lib:${LD_LIBRARY_PATH}
+                  fi
+                done
+              fi
             fi
           fi
           if [ "${INPUTS_TOTAL_WORKERS}" -gt 1 ]; then