From 376d76c25b7c949f675b3e53b77afd957471db9c Mon Sep 17 00:00:00 2001 From: Ricardo Vieira <28983449+ricardoV94@users.noreply.github.com> Date: Thu, 29 Aug 2024 15:39:44 +0200 Subject: [PATCH] Make zip strict in `apply_function_over_dataset` --- pymc/backends/arviz.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pymc/backends/arviz.py b/pymc/backends/arviz.py index a623cc0366..d1c27b787b 100644 --- a/pymc/backends/arviz.py +++ b/pymc/backends/arviz.py @@ -656,7 +656,7 @@ def apply_function_over_dataset( for idx in indices: out = fn(posterior_pts[idx]) fn.f.trust_input = True # If we arrive here the dtypes are valid - for var_name, val in zip(output_var_names, out): + for var_name, val in zip(output_var_names, out, strict=True): out_dict.insert(var_name, val, idx) progress.advance(task)