Merge pull request #22766 from jakevdp:array-api-cleanup

PiperOrigin-RevId: 657775853
jax-ml · Jul 30, 2024 · 7583cbb · 7583cbb
2 parents 9c669a7 + 1259322
commit 7583cbb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 53 deletions.
diff --git a/jax/experimental/array_api/__init__.py b/jax/experimental/array_api/__init__.py
@@ -72,6 +72,7 @@
     broadcast_to as broadcast_to,
     can_cast as can_cast,
     ceil as ceil,
+    clip as clip,
     complex128 as complex128,
     complex64 as complex64,
     concat as concat,
@@ -100,6 +101,7 @@
     full_like as full_like,
     greater as greater,
     greater_equal as greater_equal,
+    hypot as hypot,
     iinfo as iinfo,
     imag as imag,
     inf as inf,
@@ -195,8 +197,3 @@
 from jax.experimental.array_api._data_type_functions import (
     astype as astype,
 )
-
-from jax.experimental.array_api._elementwise_functions import (
-    clip as clip,
-    hypot as hypot,
-)
diff --git a/jax/experimental/array_api/_elementwise_functions.py b/jax/experimental/array_api/_elementwise_functions.py