remove deprecated jnp.round_

chaobrain · Dec 14, 2024 · 18f78fb · 18f78fb
1 parent 86a9689
commit 18f78fb
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/brainunit/math/_fun_keep_unit.py b/brainunit/math/_fun_keep_unit.py
@@ -3393,13 +3393,12 @@ def round_(
     -------
     out : jax.Array
     """
-    return _fun_keep_unit_unary(jnp.round_, x)
+    return _fun_keep_unit_unary(jnp.round, x)
 
 
 @set_module_as('brainunit.math')
-def around(
+def round(
     x: Union[Quantity, jax.typing.ArrayLike],
-    decimals: int = 0,
 ) -> jax.Array | Quantity:
     """
     Round an array to the nearest integer.
@@ -3408,18 +3407,16 @@ def around(
     ----------
     x : array_like, Quantity
       Input array.
-    decimals : int, optional
-      Number of decimal places to round to (default is 0).
 
     Returns
     -------
     out : jax.Array
     """
-    return _fun_keep_unit_unary(jnp.around, x, decimals=decimals)
+    return _fun_keep_unit_unary(jnp.round, x)
 
 
 @set_module_as('brainunit.math')
-def round(
+def around(
     x: Union[Quantity, jax.typing.ArrayLike],
     decimals: int = 0,
 ) -> jax.Array | Quantity:
@@ -3437,7 +3434,7 @@ def round(
     -------
     out : jax.Array
     """
-    return _fun_keep_unit_unary(jnp.round, x, decimals=decimals)
+    return _fun_keep_unit_unary(jnp.around, x, decimals=decimals)
 
 
 @set_module_as('brainunit.math')