Update random.py

brainpy · Aug 3, 2024 · da498b5 · da498b5
1 parent 21068ff
commit da498b5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/brainpy/_src/math/random.py b/brainpy/_src/math/random.py
@@ -67,10 +67,9 @@ def _size2shape(size):
 
 
 def _check_shape(name, shape, *param_shapes):
-  shape = core.as_named_shape(shape)
   if param_shapes:
-    shape_ = lax.broadcast_shapes(shape.positional, *param_shapes)
-    if shape.positional != shape_:
+    shape_ = lax.broadcast_shapes(shape, *param_shapes)
+    if shape != shape_:
       msg = ("{} parameter shapes must be broadcast-compatible with shape "
              "argument, and the result of broadcasting the shapes must equal "
              "the shape argument, but got result {} for shape argument {}.")