Skip to content

Commit 36db815

Browse files
Update Arviz conversion for new missing values approach
1 parent fb21db8 commit 36db815

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

pymc3/backends/arviz.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919

2020
from aesara.graph.basic import Constant
2121
from aesara.tensor.sharedvar import SharedVariable
22+
from aesara.tensor.subtensor import AdvancedIncSubtensor
2223
from arviz import InferenceData, concat, rcParams
2324
from arviz.data.base import CoordSpec, DimSpec
2425
from arviz.data.base import dict_to_dataset as _dict_to_dataset
@@ -275,9 +276,10 @@ def log_likelihood_vals_point(self, point, var, log_like_fun):
275276
# variables some other way
276277
point = {i.name: point[i.name] for i in log_like_fun.f.maker.inputs if i.name in point}
277278
log_like_val = np.atleast_1d(log_like_fun(point))
278-
if var.tag.missing_values:
279+
280+
if isinstance(var.owner.op, AdvancedIncSubtensor):
279281
try:
280-
obs_data = extract_obs_data(var)
282+
obs_data = extract_obs_data(var.tag.observations)
281283
except TypeError:
282284
warnings.warn(f"Could not extract data from symbolic observation {var}")
283285

0 commit comments

Comments
 (0)