Update jaxlib for evofr on linux/arm64 #171
Conversation
There was a problem hiding this comment.
(non-blocking)
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.
I'm glad CI caught this, but still, it probably would have taken me some time to figure out that updating jaxlib is the solution since I'm not familiar with evofr.
I can't think of a good way to accommodate future occurrences, barring official pre-built binaries for linux/arm64. Maybe a note alongside the evofr installation hinting at a pinned jaxlib would help? This also applies to Augur's CVXOPT dependency, which is now pinned.
There was a problem hiding this comment.
Maybe a note alongside the evofr installation hinting at a pinned jaxlib would help?
Good idea! I added a note for our future selves in the latest commit.
There was a problem hiding this comment.
I was wondering if there was an option for pip to warn us if it's unable to install the latest version because it's unable to satisfy dependency requirements. I found some discussions in pypa/pip#10784 for potentially adding a new feature for warning on mismatched Python version requirements. However, I can't seem to find an existing option to avoid this silent downgrade 😞
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. ¹ blab/evofr@46f744b ² jax-ml/jax#7097 ³ https://github.com/nextstrain/docker-base/actions/runs/5480222698
Description of proposed changes
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.
¹ blab/evofr@46f744b
² jax-ml/jax#7097
³ https://github.com/nextstrain/docker-base/actions/runs/5480222698
Testing