diff --git a/pymc_experimental/inference/pathfinder.py b/pymc_experimental/inference/pathfinder.py index 1092341a0..85c1981ee 100644 --- 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( diff --git a/pytest.ini b/pytest.ini new file mode 100644 index 000000000..ba9c51adb --- /dev/null +++ b/pytest.ini @@ -0,0 +1,6 @@ +[pytest] +filterwarnings = + error + ignore:.*?(\b(pkg_resources\.declare_namespace)\b).*:DeprecationWarning + ignore::UserWarning:arviz.data.inference_data + ignore::DeprecationWarning:pkg_resources