Solve for pymc>=5.16.0 (#485)

arviz-devs · Jun 29, 2024 · 943c697 · 943c697
1 parent 97ecb7e
commit 943c697
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/preliz/ppls/pymc_io.py b/preliz/ppls/pymc_io.py
@@ -204,8 +204,8 @@ def reshape_params(model, var_info, p_model, params):
 def non_constant_parents(var_):
     """Find the parents of a variable that are not constant."""
     parents = []
-    for variable in var_.get_parents()[0].inputs[3:]:
+    for variable in var_.get_parents()[0].inputs[2:]:
         if not isinstance(variable, TensorConstant):
-            parents.append(variable)
+            parents.append(variable.owner.inputs[0])
 
     return parents
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -9,5 +9,5 @@ nbconvert===6.5.4
 nbformat==5.1.3
 lxml==5.0.0
 ipykernel
-pymc<=5.15.1
+pymc>=5.16.0
 bambi>=0.13.0