metropolis: time_per_sample ensure order in point

hvasbath · Sep 18, 2024 · 114c855 · 114c855
1 parent 4d23f8f
commit 114c855
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/beat/sampler/metropolis.py b/beat/sampler/metropolis.py
@@ -260,7 +260,10 @@ def time_per_sample(self, n_points=10):
         if not self._tps:
             tps = num.zeros((n_points))
             for i in range(n_points):
-                q = self.bij.map(self.population[i])
+                point = self.population[i]
+                point = {val_var.name: point[val_var.name] for val_var in self.value_vars}
+                q = self.bij.map(point)
+
                 t0 = time()
                 self.logp_forw_func(q.data)
                 t1 = time()