Update dependencies - Githubissues

JaxGaussianProcesses / GPJax

Gaussian processes in JAX.

https://docs.jaxgaussianprocesses.com/

Apache License 2.0

436 stars 51 forks source link

Update dependencies #442

Closed Thomas-Christie closed 6 months ago

Thomas-Christie commented 6 months ago

Type of changes

[ ] Bug fix
[ ] New feature
[ ] Documentation / docstrings
[ ] Tests
[X] Other

Checklist

[ ] I've formatted the new code by running poetry run pre-commit run --all-files --show-diff-on-failure before committing.
[N/a] I've added tests for new code.
[N/a] I've added docstrings for the new code.

Description

Fixed issue #397

Bumped tensorflow-probability to ^0.22.0 and fixed issues arising as a result of this.
As part of this, updated to using new JAX random keys.
Removed benchmarks/linops.py.
Also fixed issue mentioned in PR #437

Issue Number: #397