Add isotonic module.

vroulet · Feb 16, 2023 · f65001b · f65001b
1 parent 0472831
commit f65001b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/jaxopt/__init__.py b/jaxopt/__init__.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 from jaxopt import implicit_diff
+from jaxopt import isotonic
 from jaxopt import loss
 from jaxopt import objective
 from jaxopt import projection

diff --git a/tests/import_test.py b/tests/import_test.py
@@ -24,6 +24,10 @@ def test_implicit_diff(self):
     jaxopt.implicit_diff.root_vjp
     from jaxopt.implicit_diff import root_vjp
 
+  def test_isotonic(self):
+    jaxopt.isotonic.isotonic_l2_pav
+    from jaxopt.isotonic import isotonic_l2_pav
+
   def test_prox(self):
     jaxopt.prox.prox_none
     from jaxopt.prox import prox_none