Handle some other distributions which have r_v.owner.op.name = None (#…

…516)
arviz-devs · Aug 13, 2024 · 9c7bde3 · 9c7bde3
1 parent a8d20b7
commit 9c7bde3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/preliz/ppls/pymc_io.py b/preliz/ppls/pymc_io.py
@@ -123,8 +123,9 @@ def get_model_information(model):  # pylint: disable=too-many-locals
         size = r_v_eval.size
         shape = r_v_eval.shape
         nc_parents = non_constant_parents(r_v, model.free_RVs)
-
-        name = r_v.owner.op.name
+        name = (
+            r_v.owner.op.name if r_v.owner.op.name else str(r_v.owner.op).split("RV", 1)[0].lower()
+        )
         dist = pymc_to_preliz[name]
         p_model[r_v.name] = dist
         if nc_parents: