From 7136bc46e8a925a06f4604a72e1eb0f652246051 Mon Sep 17 00:00:00 2001 From: rohanbabbar04 Date: Tue, 13 Aug 2024 11:13:32 +0530 Subject: [PATCH] Handle some other distributions which have r_v.owner.op.name = None --- preliz/ppls/pymc_io.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/preliz/ppls/pymc_io.py b/preliz/ppls/pymc_io.py index f88c017f..4823d819 100644 --- 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: