Fix lift_rv_shapes when size is empty and parameters are broadcasted

aesara/tensor/random/opt.py

@@ -58,14 +58,20 @@ def lift_rv_shapes(node):
 
     dist_params = broadcast_params(dist_params, node.op.ndims_params)
 
-    dist_params = [
-        broadcast_to(
-            p, (tuple(size) + tuple(p.shape)) if node.op.ndim_supp > 0 else size
-        )
-        for p in dist_params
-    ]
+    if get_vector_length(size) > 0:
+        dist_params = [
+            broadcast_to(
+                p, (tuple(size) + tuple(p.shape)) if node.op.ndim_supp > 0 else size
+            )
+            for p in dist_params
+        ]
+
+    new_node = node.op.make_node(rng, None, dtype, *dist_params)
+
+    if config.compute_test_value != "off":
+        compute_test_value(new_node)
 
-    return node.op.make_node(rng, None, dtype, *dist_params)
+    return new_node
 
 
 @local_optimizer([DimShuffle])

tests/tensor/random/test_opt.py

@@ -101,6 +101,13 @@ def test_lift_rv_shapes():
     test_size = []
     check_shape_lifted_rv(normal, test_params, test_size, rng)
 
+    test_params = [
+        np.array([0.0, 1.0], dtype=config.floatX),
+        np.array(5.0, dtype=config.floatX),
+    ]
+    test_size = []
+    check_shape_lifted_rv(normal, test_params, test_size, rng)
+
     test_params = [
         np.array([0.0, 1.0], dtype=config.floatX),
         np.array(5.0, dtype=config.floatX),