use mode=FAST_COMPILE to get unobserved_value_vars after MAP optimization

jessegrabowski · jessegrabowski · commit 2035202dd72f · 2024-12-04T19:09:07.000+08:00
diff --git a/pymc_experimental/inference/find_map.py b/pymc_experimental/inference/find_map.py
@@ -418,7 +418,7 @@ def find_MAP(
 
     raveled_optimized = RaveledVars(optimizer_result.x, initial_params.point_map_info)
     unobserved_vars = get_default_varnames(model.unobserved_value_vars, include_transformed)
-    unobserved_vars_values = model.compile_fn(unobserved_vars)(
+    unobserved_vars_values = model.compile_fn(unobserved_vars, mode="FAST_COMPILE")(
         DictToArrayBijection.rmap(raveled_optimized)
     )
 

Original file line number	Diff line number	Diff line change
`@@ -418,7 +418,7 @@ def find_MAP(`
`418`	`418`
`419`	`419`	`raveled_optimized = RaveledVars(optimizer_result.x, initial_params.point_map_info)`
`420`	`420`	`unobserved_vars = get_default_varnames(model.unobserved_value_vars, include_transformed)`
`421`		`- unobserved_vars_values = model.compile_fn(unobserved_vars)(`
	`421`	`+ unobserved_vars_values = model.compile_fn(unobserved_vars, mode="FAST_COMPILE")(`
`422`	`422`	`DictToArrayBijection.rmap(raveled_optimized)`
`423`	`423`	`)`
`424`	`424`