From 9fa64fba284abea96fc2bc5693496f696f7c73dd Mon Sep 17 00:00:00 2001 From: Jover Date: Thu, 6 Jul 2023 15:48:28 -0700 Subject: [PATCH] Update jaxlib for evofr on linux/arm64 MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The latest release of evofr included updates to its dependencies that updated the minimum version for jaxlib.¹ I'm using the latest release of jaxlib available in the user forked repo since there's still no official pre-built binaries for linux/arm64.² I did not realize this change was needed until I saw the "Validate Platform" job fail in the CI run³ that I triggered to update evofr to v0.1.20. Added a note for our future selves to check the jaxlib installation if we run into different versions of evofr on the different platforms. ¹ https://github.com/blab/evofr/commit/46f744b605c258d6f3d3b7b3ac8bc877f8e4ba22 ² https://github.com/google/jax/issues/7097 ³ https://github.com/nextstrain/docker-base/actions/runs/5480222698 --- Dockerfile | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 18a1718..d31263c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -258,7 +258,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ # linux/arm64. A GitHub user has provided them in a fork repo. # https://github.com/google/jax/issues/7097#issuecomment-1110730040 RUN if [[ "$TARGETPLATFORM" == linux/arm64 ]]; then \ - pip3 install https://github.com/yoziru/jax/releases/download/jaxlib-v0.3.25/jaxlib-0.3.25-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.whl \ + pip3 install https://github.com/yoziru/jax/releases/download/jaxlib-v0.4.6/jaxlib-0.4.6-cp310-cp310-manylinux2014_aarch64.manylinux_2_17_aarch64.whl \ ; \ fi @@ -354,6 +354,9 @@ RUN /builder-scripts/download-repo https://github.com/nextstrain/augur "$(/build && pip3 install --editable . # Add evofr for forecasting +# NOTE: if there is an issue with the evofr installation on linux/arm64, make +# sure to check that the jaxlib installation above satisfies the latest evofr +# dependency requirements. RUN pip3 install evofr # ———————————————————————————————————————————————————————————————————— #