Skip to content

Conversation

@btalamini
Copy link
Collaborator

@btalamini btalamini commented Oct 15, 2025

Update the package build to declare the minimum Python version to be 3.13. On my Mac M2 at least, this let me stop pinning old versions of Jax and Scipy. The scikit-sparse dependency was also easy - the current version of suite-sparse in Homebrew worked without any intervention from me.

I fixed all tests that failed due to the upgrade. All were due to minor changes in Jax's interface (plus the problem where it no longer seems to support auxiliary outputs from solvers in lax.custom_root, which I consider to be a bug).

@cmhamel or @ralberd, once this is passing the new py3.13 test, please remove the requirements for the 3.9 and 3.10 tests, and make the 3.13 test required for merging.

Fixes #92

Test results math to 14 digits or so, but now fall just outside the prior tolerance
Scipy was sending integer-valued arrays into jax jvp functions.
I don't know why - maybe the initial guess is a zero vector of
ints? Anyway, this guards against that by explicitly converting
back and forth between numpy and jax.numpy arrays as necessary,
with jax arrays having their dtype explicitly set to a float type.
A test is barely failing on github actions. I can't reproduce it
because my machine (which is a mac, not linux ubuntu like GH actions)
apparently uses a different random seed and has values for the test.
@btalamini btalamini force-pushed the update_build_to_python_3.13 branch from 61dedec to fd0c860 Compare October 17, 2025 20:11
@btalamini btalamini requested review from cmhamel and ralberd October 17, 2025 20:34
@codecov
Copy link

codecov bot commented Oct 17, 2025

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 76.11%. Comparing base (09ad182) to head (968180d).
⚠️ Report is 15 commits behind head on main.

Additional details and impacted files
@@            Coverage Diff             @@
##             main     #122      +/-   ##
==========================================
+ Coverage   76.06%   76.11%   +0.04%     
==========================================
  Files          66       67       +1     
  Lines        5832     5840       +8     
==========================================
+ Hits         4436     4445       +9     
+ Misses       1396     1395       -1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@cmhamel
Copy link
Collaborator

cmhamel commented Oct 17, 2025

@btalamini @ralberd I tweaked the CI and setup.py to also allow for python 3.11 and 3.12 since Ryan needs 3.11 with his current spack build. All tests are passing.

I'm going to leave this unmerged until Ryan gets a chance to try it out with his plato build.

@ralberd ralberd merged commit 44438c1 into sandialabs:main Oct 19, 2025
8 checks passed
@btalamini btalamini deleted the update_build_to_python_3.13 branch November 27, 2025 18:51
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Fix bug in jax 4.0.34

3 participants