From 087b01206466f4ba90be0c36f33f4191e5504205 Mon Sep 17 00:00:00 2001 From: Oriol Abril-Pla Date: Wed, 16 Jul 2025 17:09:16 +0200 Subject: [PATCH] Pass model to `add_fit_to_inference_data` --- pymc_extras/inference/laplace_approx/find_map.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pymc_extras/inference/laplace_approx/find_map.py b/pymc_extras/inference/laplace_approx/find_map.py index 79a1dea50..19bf7fd3a 100644 --- a/pymc_extras/inference/laplace_approx/find_map.py +++ b/pymc_extras/inference/laplace_approx/find_map.py @@ -336,7 +336,7 @@ def find_MAP( } idata = map_results_to_inference_data(optimized_point, frozen_model, include_transformed) - idata = add_fit_to_inference_data(idata, raveled_optimized, H_inv) + idata = add_fit_to_inference_data(idata, raveled_optimized, H_inv, model) idata = add_optimizer_result_to_inference_data( idata, optimizer_result, method, raveled_optimized, model )