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

Fix the ValueError message for random.binomial (forgot to use string formatting).

Browse Source
This commit is contained in:
carlosgmartin 2025-03-10 16:38:03 -04:00
parent affe2e734e
commit 8b6ca56417
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/_src/random.py
View File

@ -2646,7 +2646,7 @@ def binomial(
dtypes.check_user_dtype_supported(dtype)
if not dtypes.issubdtype(dtype, np.floating):
raise ValueError(
"dtype argument to `binomial` must be a float dtype, got {dtype}"
f"dtype argument to `binomial` must be a float dtype, got {dtype}"
)
dtype = dtypes.canonicalize_dtype(dtype)
if shape is not None: