Skip to content

Commit 16400d9

Browse files
Avoid PyMC deprecations
1 parent b647b66 commit 16400d9

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

mcbackend/adapters/pymc.py

+5-5
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,7 @@
2626

2727
def find_data(pmodel: Model) -> List[DataVariable]:
2828
"""Extracts data variables from a model."""
29-
observed_rvs = {
30-
rv.tag.observations for rv in pmodel.observed_RVs if hasattr(rv.tag, "observations")
31-
}
29+
observed_rvs = {pmodel.rvs_to_values[rv] for rv in pmodel.observed_RVs}
3230
dvars = []
3331
# All data containers are named vars!
3432
for name, var in pmodel.named_vars.items():
@@ -39,7 +37,7 @@ def find_data(pmodel: Model) -> List[DataVariable]:
3937
dv.value = ndarray_from_numpy(var.get_value())
4038
else:
4139
continue
42-
dv.dims = list(pmodel.RV_dims.get(name, []))
40+
dv.dims = list(pmodel.named_vars_to_dims.get(name, []))
4341
dv.is_observed = var in observed_rvs
4442
dvars.append(dv)
4543
return dvars
@@ -142,7 +140,9 @@ def setup(
142140
name,
143141
str(self.var_dtypes[name]),
144142
list(self.var_shapes[name]),
145-
dims=list(self.model.RV_dims[name]) if name in self.model.RV_dims else [],
143+
dims=list(self.model.named_vars_to_dims[name])
144+
if name in self.model.named_vars_to_dims
145+
else [],
146146
is_deterministic=(name not in free_rv_names),
147147
)
148148
for name in self.varnames

0 commit comments

Comments
 (0)