Merge pull request jax-ml#12308 from jakevdp:fix-shape-error

PiperOrigin-RevId: 473319485
jakevdp · Sep 9, 2022 · 056f400 · 056f400
2 parents 4746a39 + fcac395
commit 056f400
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/jax/_src/lax/lax.py b/jax/_src/lax/lax.py
@@ -149,8 +149,7 @@ def _broadcast_shapes_uncached(*shapes):
   shape_list = [(1,) * (ndim - len(shape)) + shape for shape in shapes]
   result_shape = _try_broadcast_shapes(shape_list)
   if result_shape is None:
-    raise ValueError("Incompatible shapes for broadcasting: {}"
-                     .format(tuple(shape_list)))
+    raise ValueError(f"Incompatible shapes for broadcasting: shapes={list(shapes)}")
   return result_shape
 
 def _broadcast_ranks(s1, s2):