Skip to content

Commit d5355a5

Browse files
ferrinericardoV94
andauthored
Update pymc_experimental/model/transforms/autoreparam.py
Co-authored-by: Ricardo Vieira <[email protected]>
1 parent 82a1a24 commit d5355a5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

pymc_experimental/model/transforms/autoreparam.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,7 @@ def vip_reparam_node(
176176
) -> Tuple[ModelDeterministic, ModelNamed]:
177177
if not isinstance(node.op, RandomVariable | SymbolicRandomVariable):
178178
raise TypeError("Op should be RandomVariable type")
179+
# FIXME: This is wrong when size is None
179180
_, size, *_ = node.inputs
180181
eval_size = size.eval(mode="FAST_COMPILE")
181182
if eval_size is not None:

0 commit comments

Comments
 (0)