diff --git a/pymc3/step_methods/nuts.py b/pymc3/step_methods/nuts.py index 5121681256..b72813156d 100644 --- a/pymc3/step_methods/nuts.py +++ b/pymc3/step_methods/nuts.py @@ -157,7 +157,7 @@ def astep(self, q0): self.Hbar = (1 - w) * self.Hbar + w * \ (self.target_accept - a * 1. / na) - self.step_size = exp(self.u - (self.m**.5 / self.gamma) * self.Hbar) + self.step_size = exp(self.u - (self.m**self.k / self.gamma) * self.Hbar) self.m += 1 return q