Update Op checks to include both AdvancedIncSubtensor Ops

Author: brandonwillard

pymc3/backends/arviz.py

@@ -19,7 +19,7 @@
 
 from aesara.graph.basic import Constant
 from aesara.tensor.sharedvar import SharedVariable
-from aesara.tensor.subtensor import AdvancedIncSubtensor
+from aesara.tensor.subtensor import AdvancedIncSubtensor, AdvancedIncSubtensor1
 from arviz import InferenceData, concat, rcParams
 from arviz.data.base import CoordSpec, DimSpec
 from arviz.data.base import dict_to_dataset as _dict_to_dataset
@@ -283,7 +283,7 @@ def log_likelihood_vals_point(self, point, var, log_like_fun):
         point = {i.name: point[i.name] for i in log_like_fun.f.maker.inputs if i.name in point}
         log_like_val = np.atleast_1d(log_like_fun(point))
 
-        if isinstance(var.owner.op, AdvancedIncSubtensor):
+        if isinstance(var.owner.op, (AdvancedIncSubtensor, AdvancedIncSubtensor1)):
             try:
                 obs_data = extract_obs_data(var.tag.observations)
             except TypeError:

pymc3/tests/test_idata_conversion.py

@@ -5,7 +5,7 @@
 import pandas as pd
 import pytest
 
-from aesara.tensor.subtensor import AdvancedIncSubtensor
+from aesara.tensor.subtensor import AdvancedIncSubtensor, AdvancedIncSubtensor1
 from arviz import InferenceData
 from arviz.tests.helpers import check_multiple_attrs
 from numpy import ma
@@ -327,7 +327,7 @@ def test_mv_missing_data_model(self):
             inference_data = pm.sample(100, chains=2, return_inferencedata=True)
 
         # make sure that data is really missing
-        assert isinstance(y.owner.op, AdvancedIncSubtensor)
+        assert isinstance(y.owner.op, (AdvancedIncSubtensor, AdvancedIncSubtensor1))
 
         test_dict = {
             "posterior": ["mu", "chol_cov"],