mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Finalize deprecation of jnp.round_

Browse Source 
PiperOrigin-RevId: 705998500
This commit is contained in:
Jake VanderPlas 2024-12-13 14:13:14 -08:00 committed by jax authors
parent 078c7e4444
commit c73f306099
3 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CHANGELOG.md
View File

@ -26,6 +26,7 @@ When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG.
`non_negative_dim`. `non_negative_dim`.
* from {mod}`jax.lib.xla_bridge`: `xla_client` and `default_backend`. * from {mod}`jax.lib.xla_bridge`: `xla_client` and `default_backend`.
* from {mod}`jax.lib.xla_client`: `_xla` and `bfloat16`. * from {mod}`jax.lib.xla_client`: `_xla` and `bfloat16`.
* from {mod}`jax.numpy`: `round_`.
## jax 0.4.37 (Dec 9, 2024) ## jax 0.4.37 (Dec 9, 2024)

1
docs/jax.numpy.rst
View File

@ -359,7 +359,6 @@ namespace; they are listed below.
roots roots
rot90 rot90
round round
round_
s_ s_
save save
savez savez

10
jax/numpy/__init__.py
View File

@ -479,17 +479,15 @@ del register_jax_array_methods
_deprecations = { _deprecations = {
# Deprecated 03 Sept 2024 # Finalized 2024-12-13; remove after 2024-3-13
"round_": ( "round_": (
"jnp.round_ is deprecated; use jnp.round instead.", "jnp.round_ was deprecated in JAX 0.4.38; use jnp.round instead.",
round None
), ),
} }
import typing import typing
if typing.TYPE_CHECKING: if not typing.TYPE_CHECKING:
round_ = round
else:
from jax._src.deprecations import deprecation_getattr as _deprecation_getattr from jax._src.deprecations import deprecation_getattr as _deprecation_getattr
__getattr__ = _deprecation_getattr(__name__, _deprecations) __getattr__ = _deprecation_getattr(__name__, _deprecations)
del _deprecation_getattr del _deprecation_getattr