minor edits

vroulet · Jun 7, 2023 · 8ca6e67 · 8ca6e67
1 parent 4c847ea
commit 8ca6e67
Show file tree

Hide file tree

Showing 3 changed files with 170 additions and 146 deletions.
diff --git a/jaxopt/__init__.py b/jaxopt/__init__.py
@@ -34,7 +34,6 @@
 from jaxopt._src.gauss_newton import GaussNewton
 from jaxopt._src.gradient_descent import GradientDescent
 from jaxopt._src.hager_zhang_linesearch import HagerZhangLineSearch
-from jaxopt._src.zoom_linesearch2 import ZoomLineSearch
 from jaxopt._src.iterative_refinement import IterativeRefinement
 from jaxopt._src.lbfgs import LBFGS
 from jaxopt._src.lbfgsb import LBFGSB
@@ -52,3 +51,4 @@
 from jaxopt._src.scipy_wrappers import ScipyLeastSquares
 from jaxopt._src.scipy_wrappers import ScipyMinimize
 from jaxopt._src.scipy_wrappers import ScipyRootFinding
+from jaxopt._src.zoom_linesearch2 import ZoomLineSearch