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

Remove jax.util.partial.

Browse Source
This commit is contained in:
Peter Hawkins 2021-09-20 20:32:49 -04:00
parent 9a5cf7c81e
commit 58c7ee46bc
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
CHANGELOG.md
View File

@ -15,9 +15,9 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
* `jax.api` has been removed. Functions that were available as `jax.api.*`
were aliases for functions in `jax.*`; please use the functions in
`jax.*` instead.
* `jax.partial` and `jax.lax.partial` were accidental exports that have now
been removed. Use `functools.partial` from the Python standard library
instead.
* `jax.partial`, `jax.lax.partial`, and `jax.util.partial` were accidental
exports that have now been removed. Use `functools.partial` from the Python
standard library instead.
* Boolean scalar indices now raise a `TypeError`; previously this silently
returned wrong results ({jax-issue}`#7925`).
* Many more `jax.numpy` functions now require array-like inputs, and will error

1
jax/util.py
View File

@ -18,7 +18,6 @@ from jax._src.util import (
HashableFunction as HashableFunction,
as_hashable_function as as_hashable_function,
cache as cache,
partial as partial,
safe_map as safe_map,
safe_zip as safe_zip,
split_dict as split_dict,