search

JaxGaussianProcesses / GPJax

Gaussian processes in JAX.
https://docs.jaxgaussianprocesses.com/
Apache License 2.0
436 stars 51 forks source link

docs: Missing Dependency: "jaxopt" Not Listed in gpjax Documentation #444

Closed LiHao-MS closed 5 months ago

LiHao-MS commented 5 months ago

When attempting to use “gpjax” by importing the fit function via “from gpjax.fit import fit”, an ImportError occurs if jaxopt is not installed: cannot import name 'ScipyBoundedMinimize' from 'jaxopt' (unknown location). However, jaxopt is not listed in the requirements for gpjax. This issue traces back to ./gpjax/decision_making/utility_maximizer.py, specifically at line 23: from jaxopt import ScipyBoundedMinimize.

Considering that “jaxopt” has transitioned to maintenance mode and its functionalities are being integrated into “Optax”, it might be advantageous to replace all code dependencies on “jaxopt” with their “Optax” counterparts. This shift could significantly improve the future compatibility and maintainability of "gpjax".

thomaspinder commented 5 months ago

Thanks for raising this. Fixed now in v0.8.2.