feat: arctan2 (#37)

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Mar 27, 2024 · 8636018 · 8636018
1 parent ea8be18
commit 8636018
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/quaxed/numpy/_core.py b/src/quaxed/numpy/_core.py
@@ -4,6 +4,7 @@
     "allclose",
     "array_equal",
     "asarray",
+    "arctan2",
     "cbrt",
     "copy",
     "cross",
@@ -60,6 +61,7 @@ def transfer_doc(func: T) -> T:
 allclose = quaxify(jnp.allclose)
 array_equal = quaxify(jnp.array_equal)
 asarray = quaxify(jnp.asarray)
+arctan2 = quaxify(jnp.arctan2)
 cbrt = quaxify(jnp.cbrt)
 copy = quaxify(jnp.copy)
 cross = quaxify(jnp.cross)