Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Relax Jax version #103

Merged
merged 4 commits into from
Aug 23, 2022
Merged

Relax Jax version #103

merged 4 commits into from
Aug 23, 2022

Conversation

thomaspinder
Copy link
Collaborator

This PR relaxes the strict Jax requirement present in previous version of GPJax. This should resolve #102

Pull request type

Please check the type of change your PR introduces:

  • [ x ] Bugfix
  • Feature
  • Code style update (formatting, renaming)
  • Refactoring (no functional changes, no api changes)
  • Build related changes
  • Documentation content changes
  • Other (please describe):

What is the current behavior?

Right now we cannot use Jax versions beyond 0.3.5. By extending the distrax bijectors and updating some Jax tree-based operations, we can simply make this versioning an inequality to install the latest Jax version.

Issue Number: #102

@codecov
Copy link

codecov bot commented Aug 23, 2022

Codecov Report

Merging #103 (2e5be70) into master (2f00b11) will decrease coverage by 0.00%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##           master     #103      +/-   ##
==========================================
- Coverage   98.96%   98.95%   -0.01%     
==========================================
  Files          13       13              
  Lines         962      960       -2     
==========================================
- Hits          952      950       -2     
  Misses         10       10              
Flag Coverage Δ
unittests 98.95% <100.00%> (-0.01%) ⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files Coverage Δ
gpjax/config.py 100.00% <100.00%> (ø)
gpjax/parameters.py 96.00% <100.00%> (-0.07%) ⬇️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

@thomaspinder
Copy link
Collaborator Author

@daniel-dodd - can I get a quick review on this please? There's no frontend changes, it simply lets us relax the Jax version.

gpjax/parameters.py Outdated Show resolved Hide resolved
Copy link
Member

@daniel-dodd daniel-dodd left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@thomaspinder This PR looks good to me.

For consistency, please rename "unit-Gaussian" to "unit Gaussian" in the warning message.

@daniel-dodd daniel-dodd merged commit 4fca81e into master Aug 23, 2022
@thomaspinder thomaspinder deleted the fix_deps branch August 23, 2022 09:42
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.

bug: Installation with GPU support
2 participants