From d366e128f24092b2d1818b4da5fb9cf0fe90b837 Mon Sep 17 00:00:00 2001 From: Rob Zinkov Date: Mon, 1 Jan 2024 17:55:21 +0100 Subject: [PATCH] Remove use of treedict, addresses #291 --- pymc_experimental/model/marginal_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pymc_experimental/model/marginal_model.py b/pymc_experimental/model/marginal_model.py index 1eb23ff2..1f8e4531 100644 --- a/pymc_experimental/model/marginal_model.py +++ b/pymc_experimental/model/marginal_model.py @@ -84,7 +84,7 @@ class MarginalModel(Model): def __init__(self, *args, **kwargs): super().__init__(*args, **kwargs) self.marginalized_rvs = [] - self._marginalized_named_vars_to_dims = treedict() + self._marginalized_named_vars_to_dims = {} def _delete_rv_mappings(self, rv: TensorVariable) -> None: """Remove all model mappings referring to rv