Skip to content

Commit 60a04ff

Browse files
committed
fix: do not trigger C compile when evaluating shape
1 parent 057a695 commit 60a04ff

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pymc_experimental/model/transforms/autoreparam.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,7 @@ def vip_reparam_node(
177177
if not isinstance(node.op, RandomVariable | SymbolicRandomVariable):
178178
raise TypeError("Op should be RandomVariable type")
179179
_, size, *_ = node.inputs
180-
eval_size = size.eval()
180+
eval_size = size.eval(mode="FAST_COMPILE")
181181
if eval_size is not None:
182182
rv_shape = tuple(eval_size)
183183
else:

0 commit comments

Comments
 (0)