fixing pathfinder deprecated model.rv_dim call

michaelraczycki · michaelraczycki · commit c7a86ff6dead · 2023-04-10T09:18:48.000+01:00
diff --git a/pymc_experimental/inference/pathfinder.py b/pymc_experimental/inference/pathfinder.py
@@ -138,7 +138,7 @@ def one_step(state, rng_key):
 
     dims = {
         var_name: [dim for dim in dims if dim is not None]
-        for var_name, dims in model.RV_dims.items()
+        for var_name, dims in model.named_vars_to_dims.items()
     }
 
     idata = convert_flat_trace_to_idata(

Original file line number	Diff line number	Diff line change
`@@ -138,7 +138,7 @@ def one_step(state, rng_key):`
`138`	`138`
`139`	`139`	`dims = {`
`140`	`140`	`var_name: [dim for dim in dims if dim is not None]`
`141`		`- for var_name, dims in model.RV_dims.items()`
	`141`	`+ for var_name, dims in model.named_vars_to_dims.items()`
`142`	`142`	`}`
`143`	`143`
`144`	`144`	`idata = convert_flat_trace_to_idata(`