Fix bug

chaobrain · Dec 22, 2024 · 3ff33c7 · 3ff33c7
1 parent 1a6ab3e
commit 3ff33c7
Showing 1 changed file with 1 addition and 19 deletions.
diff --git a/brainunit/math/_fun_keep_unit.py b/brainunit/math/_fun_keep_unit.py
@@ -59,7 +59,7 @@
     'percentile', 'nanpercentile', 'quantile', 'nanquantile',
 
     # math funcs only accept unitless (unary) can return Quantity
-    'round', 'around', 'round_', 'rint', 'floor', 'ceil', 'trunc', 'fix', 'modf',
+    'round', 'around', 'rint', 'floor', 'ceil', 'trunc', 'fix', 'modf',
 
     # math funcs keep unit (binary)
     'fmod', 'mod', 'copysign', 'remainder',
@@ -3377,24 +3377,6 @@ def unique(
                           fill_value=fill_value)
 
 
-@set_module_as('brainunit.math')
-def round_(
-    x: Union[Quantity, jax.typing.ArrayLike],
-) -> jax.Array | Quantity:
-    """
-    Round an array to the nearest integer.
-
-    Parameters
-    ----------
-    x : array_like, Quantity
-      Input array.
-
-    Returns
-    -------
-    out : jax.Array
-    """
-    return _fun_keep_unit_unary(jnp.round, x)
-
 
 @set_module_as('brainunit.math')
 def round(