Merge pull request #27041 from carlosgmartin:fix_binomial_value_error

Google-ML-Automation · Google-ML-Automation · commit e7d10a231064 · 2025-03-12T10:05:18.000-07:00
PiperOrigin-RevId: 736171463
diff --git a/jax/_src/random.py b/jax/_src/random.py
@@ -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:

Original file line number	Diff line number	Diff line change
`@@ -2646,7 +2646,7 @@ def binomial(`
`2646`	`2646`	`dtypes.check_user_dtype_supported(dtype)`
`2647`	`2647`	`if not dtypes.issubdtype(dtype, np.floating):`
`2648`	`2648`	`raise ValueError(`
`2649`		- "dtype argument to `binomial` must be a float dtype, got {dtype}"
	`2649`	+ f"dtype argument to `binomial` must be a float dtype, got {dtype}"
`2650`	`2650`	`)`
`2651`	`2651`	`dtype = dtypes.canonicalize_dtype(dtype)`
`2652`	`2652`	`if shape is not None:`