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

Fix inaccurate type annotations

Browse Source
This commit is contained in:
Jake VanderPlas 2021-10-15 15:35:43 -07:00
parent 3377abaef4
commit afe7e194e9
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
jax/_src/dtypes.py
View File

@ -40,10 +40,10 @@ _bfloat16_dtype: np.dtype = np.dtype(bfloat16)
# Default types.
bool_ = np.bool_
int_: np.dtype = np.int64 # type: ignore
float_: np.dtype = np.float64 # type: ignore
complex_ = np.complex128
bool_: type = np.bool_
int_: type = np.int64
float_: type = np.float64
complex_: type = np.complex128
# TODO(phawkins): change the above defaults to:
# int_ = np.int32