Skip to content

Commit 481342a

Browse files
accept_rate_iter should always be an array
1 parent eac0b79 commit 481342a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

pymc/step_methods/metropolis.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -308,13 +308,12 @@ def astep(self, q0: RaveledVars) -> tuple[RaveledVars, StatsType]:
308308
else:
309309
accept_rate = self.delta_logp(q, q0d)
310310
q, accepted = metrop_select(accept_rate, q, q0d, rng=self.rng)
311-
self.accept_rate_iter = accept_rate
311+
self.accept_rate_iter[:] = accept_rate
312312
self.accepted_iter[0] = accepted
313313
self.accepted_sum += accepted
314314

315315
self.steps_until_tune -= 1
316316

317-
self.accept_rate_iter = np.atleast_1d(self.accept_rate_iter)
318317
log_N = np.log(self.accept_rate_iter.shape[0])
319318

320319
stats = {

0 commit comments

Comments
 (0)