.wip

Author: ricardoV94

pytensor/tensor/rewriting/numba.py

@@ -53,7 +53,7 @@ def introduce_explicit_core_shape_blockwise(fgraph, node):
             #  ← dirichlet_rv{"(a)->(a)"}.1 [id F]
             #     └─ ···
     """
-    op: Blockwise = node.op  # type: ignore[annotation-unchecked]
+    op: Blockwise = node.op
     batch_ndim = op.batch_ndim(node)
 
     shape_feature: ShapeFeature | None = getattr(fgraph, "shape_feature", None)  # type: ignore[annotation-unchecked]
@@ -63,6 +63,7 @@ def introduce_explicit_core_shape_blockwise(fgraph, node):
             for out in node.outputs
         ]
     else:
+        # input_shape = op.infer_shape(None, node, input_shapes)
         raise ValueError
         core_shapes = op._supp_shape_from_params(op.dist_params(node))