From 62df7c3767ad5d1417ce7394fe57caf6b98025db Mon Sep 17 00:00:00 2001 From: aloctavodia Date: Tue, 7 May 2024 09:44:06 -0300 Subject: [PATCH] ignore transformation distributions when mapping pymc to preliz distributions --- preliz/ppls/pymc_io.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/preliz/ppls/pymc_io.py b/preliz/ppls/pymc_io.py index b286ba9b..c01efd01 100644 --- a/preliz/ppls/pymc_io.py +++ b/preliz/ppls/pymc_io.py @@ -68,7 +68,11 @@ def get_pymc_to_preliz(): """ Generate dictionary mapping pymc to preliz distributions """ - all_distributions = modules["preliz.distributions"].__all__ + all_distributions = [ + dist + for dist in modules["preliz.distributions"].__all__ + if dist not in ["Truncated", "Censored", "Hurdle"] + ] pymc_to_preliz = dict( zip([dist.lower() for dist in all_distributions], get_distributions(all_distributions)) )