numpy < 2.0.0 in jaxlib installation

NVIDIA · Jun 18, 2024 · d85b60a · d85b60a
1 parent 9920d90
commit d85b60a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/container/Dockerfile.jax b/.github/container/Dockerfile.jax
@@ -34,7 +34,7 @@ RUN --mount=type=ssh \
     --mount=type=secret,id=SSH_KNOWN_HOSTS,target=/root/.ssh/known_hosts \
 <<"EOF" bash -ex
     git-clone.sh ${URLREF_JAX} ${SRC_PATH_JAX}
-    sed 's/^numpy.*/numpy<2.0.0/' ${SRC_PATH_JAX}/build/requirements.txt
+    sed 's/^numpy.*/numpy<2.0.0/' ${SRC_PATH_JAX}/build/requirements.in
     git-clone.sh ${URLREF_XLA} ${SRC_PATH_XLA}
 EOF