Skip to content

Commit

Permalink
Update _compat_numpy_funcs_accept_unitless.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Routhleck committed Jun 13, 2024
1 parent 811c9f5 commit 982a8a0
Showing 1 changed file with 4 additions and 8 deletions.
12 changes: 4 additions & 8 deletions brainunit/math/_compat_numpy_funcs_accept_unitless.py
Original file line number Diff line number Diff line change
Expand Up @@ -614,7 +614,6 @@ def percentile(
overwrite_input: Optional[bool] = None,
method: str = 'linear',
keepdims: Optional[bool] = False,
interpolation: None = None,
) -> Array:
"""
Compute the q-th percentile of the data along the specified axis.
Expand Down Expand Up @@ -666,7 +665,7 @@ def percentile(
Output array.
"""
return funcs_only_accept_unitless_binary(jnp.percentile, a, q, axis=axis, out=out, overwrite_input=overwrite_input,
method=method, keepdims=keepdims, interpolation=interpolation)
method=method, keepdims=keepdims)


@set_module_as('brainunit.math')
Expand All @@ -678,7 +677,6 @@ def nanpercentile(
overwrite_input: Optional[bool] = None,
method: str = 'linear',
keepdims: Optional[bool] = False,
interpolation: None = None,
) -> Array:
"""
Compute the q-th percentile of the data along the specified axis, while ignoring nan values.
Expand Down Expand Up @@ -730,7 +728,7 @@ def nanpercentile(
Output array.
"""
return funcs_only_accept_unitless_binary(jnp.nanpercentile, a, q, axis=axis, out=out, overwrite_input=overwrite_input,
method=method, keepdims=keepdims, interpolation=interpolation)
method=method, keepdims=keepdims)


@set_module_as('brainunit.math')
Expand All @@ -742,7 +740,6 @@ def quantile(
overwrite_input: Optional[bool] = None,
method: str = 'linear',
keepdims: Optional[bool] = False,
interpolation: None = None,
) -> Array:
"""
Compute the q-th percentile of the data along the specified axis.
Expand Down Expand Up @@ -794,7 +791,7 @@ def quantile(
Output array.
"""
return funcs_only_accept_unitless_binary(jnp.quantile, a, q, axis=axis, out=out, overwrite_input=overwrite_input,
method=method, keepdims=keepdims, interpolation=interpolation)
method=method, keepdims=keepdims)


@set_module_as('brainunit.math')
Expand All @@ -806,7 +803,6 @@ def nanquantile(
overwrite_input: Optional[bool] = None,
method: str = 'linear',
keepdims: Optional[bool] = False,
interpolation: None = None,
) -> Array:
"""
Compute the q-th percentile of the data along the specified axis, while ignoring nan values.
Expand Down Expand Up @@ -858,4 +854,4 @@ def nanquantile(
Output array.
"""
return funcs_only_accept_unitless_binary(jnp.nanquantile, a, q, axis=axis, out=out, overwrite_input=overwrite_input,
method=method, keepdims=keepdims, interpolation=interpolation)
method=method, keepdims=keepdims)

0 comments on commit 982a8a0

Please sign in to comment.