Skip to content

Commit

Permalink
Ignore requirements.in
Browse files Browse the repository at this point in the history
  • Loading branch information
olupton committed May 22, 2024
1 parent 69bc7bb commit 5570f05
Showing 1 changed file with 3 additions and 6 deletions.
9 changes: 3 additions & 6 deletions .github/container/Dockerfile.jax
Original file line number Diff line number Diff line change
Expand Up @@ -86,13 +86,10 @@ COPY --from=builder /opt/manifest.d/git-clone.yaml /opt/manifest.d/git-clone.yam
ADD build-jax.sh local_cuda_arch test-jax.sh /usr/local/bin/

RUN mkdir -p /opt/pip-tools.d
# Use the same requirements as JAX's hermetic Python configuration, but use an
# editable installation of jaxlib -- bazel won't accept that for the hermetic
# test environment

## Editable installations of jax and jaxlib
RUN <<"EOF" bash -ex
cp "${SRC_PATH_JAX}/build/requirements.in" /opt/pip-tools.d/requirements-jax.in
sed -i -e "s#-r test-requirements.txt##" /opt/pip-tools.d/requirements-jax.in
sed -i -e "s#jaxlib @ file://${BUILD_PATH_JAXLIB}#-e ${BUILD_PATH_JAXLIB}#" /opt/pip-tools.d/requirements-jax.in
echo "-e file://${BUILD_PATH_JAXLIB}" > /opt/pip-tools.d/requirements-jax.in
echo "-e file://${SRC_PATH_JAX}" >> /opt/pip-tools.d/requirements-jax.in
EOF

Expand Down

0 comments on commit 5570f05

Please sign in to comment.