Pass model to add_fit_to_inference_data

OriolAbril · web-flow · commit 087b01206466 · 2025-07-16T17:09:16.000+02:00
diff --git 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
     )

Original file line number	Diff line number	Diff line change
`@@ -336,7 +336,7 @@ def find_MAP(`
`336`	`336`	`}`
`337`	`337`
`338`	`338`	`idata = map_results_to_inference_data(optimized_point, frozen_model, include_transformed)`
`339`		`- idata = add_fit_to_inference_data(idata, raveled_optimized, H_inv)`
	`339`	`+ idata = add_fit_to_inference_data(idata, raveled_optimized, H_inv, model)`
`340`	`340`	`idata = add_optimizer_result_to_inference_data(`
`341`	`341`	`idata, optimizer_result, method, raveled_optimized, model`
`342`	`342`	`)`