We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0ca9f68 commit 5b0a796Copy full SHA for 5b0a796
pymc_experimental/model/transforms/autoreparam.py
@@ -176,8 +176,12 @@ def vip_reparam_node(
176
) -> Tuple[ModelDeterministic, ModelNamed]:
177
if not isinstance(node.op, RandomVariable | SymbolicRandomVariable):
178
raise TypeError("Op should be RandomVariable type")
179
- rv = node.default_output()
180
- rv_shape = rv.shape.eval(mode="FAST_COMPILE")
+ _, size, *_ = node.inputs
+ eval_size = size.eval(mode="FAST_COMPILE")
181
+ if eval_size is not None:
182
+ rv_shape = tuple(eval_size)
183
+ else:
184
+ rv_shape = ()
185
lam_name = f"{name}::lam_logit__"
186
_log.debug(f"Creating {lam_name} with shape of {rv_shape}")
187
logit_lam_ = pytensor.shared(
0 commit comments