From 9aa232f27c33a6a5a4e9acac20d52cb603469cf9 Mon Sep 17 00:00:00 2001 From: Asif Zubair Date: Thu, 24 Jul 2025 14:37:17 -0500 Subject: [PATCH 1/2] Fix for #7369: return scalar for sampler stats --- pymc/step_methods/metropolis.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pymc/step_methods/metropolis.py b/pymc/step_methods/metropolis.py index 85107cd30..ff67dffa0 100644 --- a/pymc/step_methods/metropolis.py +++ b/pymc/step_methods/metropolis.py @@ -1091,7 +1091,7 @@ class DEMetropolisZ(ArrayStepShared): "accept": (np.float64, []), "accepted": (bool, []), "tune": (bool, []), - "scaling": (np.float64, []), + "scaling": (np.float64, [None]), "lambda": (np.float64, []), } @@ -1214,7 +1214,7 @@ def astep(self, q0: RaveledVars) -> tuple[RaveledVars, StatsType]: stats = { "tune": self.tune, - "scaling": self.scaling, + "scaling": np.mean(self.scaling), "lambda": self.lamb, "accept": np.exp(accept), "accepted": accepted, From 405fd5964f59af79bcf82286b8264192a854b717 Mon Sep 17 00:00:00 2001 From: Asif Zubair Date: Sun, 27 Jul 2025 10:13:41 -0500 Subject: [PATCH 2/2] Fix for #7369: unroll change in scaling dtype --- pymc/step_methods/metropolis.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pymc/step_methods/metropolis.py b/pymc/step_methods/metropolis.py index ff67dffa0..c042bc1f3 100644 --- a/pymc/step_methods/metropolis.py +++ b/pymc/step_methods/metropolis.py @@ -1091,7 +1091,7 @@ class DEMetropolisZ(ArrayStepShared): "accept": (np.float64, []), "accepted": (bool, []), "tune": (bool, []), - "scaling": (np.float64, [None]), + "scaling": (np.float64, []), "lambda": (np.float64, []), }