Skip to content

Commit 346bd61

Browse files
aloctavodiaricardoV94
authored andcommitted
bug fix assigned variable nuts
1 parent b18a30f commit 346bd61

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pymc3/step_methods/hmc/base_hmc.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,8 @@ def __init__(
101101
# XXX: If the dimensions of these terms change, the step size
102102
# dimension-scaling should change as well, no?
103103
test_point = self._model.initial_point
104-
continuous_vars = [test_point[v.name] for v in self._model.cont_vars]
105-
size = sum(v.size for v in continuous_vars)
104+
nuts_vars = [test_point[v.name] for v in vars]
105+
size = sum(v.size for v in nuts_vars)
106106

107107
self.step_size = step_scale / (size ** 0.25)
108108
self.step_adapt = step_sizes.DualAverageAdaptation(

0 commit comments

Comments
 (0)