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

[NumPy] Remove references to deprecated NumPy type aliases.

Browse Source 
This change replaces references to a number of deprecated NumPy type aliases (np.bool, np.int, np.float, np.complex, np.object, np.str) with their recommended replacement (bool, int, float, complex, object, str).

NumPy 1.24 drops the deprecated aliases, so we must remove uses before updating NumPy.

PiperOrigin-RevId: 496691463
This commit is contained in:
Peter Hawkins 2022-12-20 10:16:31 -08:00 committed by jax authors
parent 71b896893a
commit 843bc43790
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jax/experimental/jax2tf/tests/models_test_main.py
View File

@ -138,7 +138,7 @@ def _get_random_data(x: jnp.ndarray) -> np.ndarray:
return np.random.randint(0, 100, size=x.shape, dtype=dtype)
elif np.issubdtype(dtype, np.floating):
return np.array(np.random.uniform(size=x.shape), dtype=dtype)
elif dtype == np.bool:
elif dtype == bool:
return np.random.choice(a=[False, True], size=x.shape)
else:
raise ValueError(f"Unsupported dtype for numerical comparison: {dtype}")