Skip to content

Conversation

@willGraham01
Copy link
Collaborator

Resolves #57 |

distrax looks long-abandoned, so we should jump ship too. This PR removes distrax as a requirement for the testing suite, which in turn lets us un-pin the jax version.

As a note;

  • DistributionFamily has only a single test, that uses distrax. In light of Remove the DistributionFamily class. #48, the PR simply removes this test entirely, but we should hold off hitting merge until the Remove the DistributionFamily class. #48 is solved.
  • Distribution had a single test that used distrax as a backend. This test has been edited so that, rather than comparing two different backends, the more sensible approach of comparing the Distribution implementation with the result we expect had we used the backend package directly.

pyproject.toml has had the jax version unpinned now that distrax is gone.

This class will be removed from the codebase anyway, so there is no point preserving the tests for it.
However, we should hold off merging this branch until DistributionFamily is well and truly gone.
@willGraham01 willGraham01 requested a review from mscroggs August 14, 2025 07:53
@mscroggs mscroggs merged commit 3265e61 into main Aug 14, 2025
5 checks passed
@mscroggs mscroggs deleted the wgraham/remove-distrax branch August 14, 2025 08:14
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.

Remove pinning of jax version

3 participants