From f79729cd2ed3d4d55437db1e77ab04660c6de370 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Faria?= <joao.faria@astro.up.pt>
Date: Tue, 15 Nov 2022 17:35:24 +0000
Subject: [PATCH] fix parsing of Rayleigh distribution (#76)

---
 pykima/utils.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/pykima/utils.py b/pykima/utils.py
index e604d9a..3a96513 100644
--- a/pykima/utils.py
+++ b/pykima/utils.py
@@ -350,6 +350,7 @@ def _prior_to_dist():
         'Kumaraswamy': kumaraswamy,
         'Laplace': stats.laplace,
         'Cauchy': stats.cauchy,
+        'Rayleigh': lambda scale: stats.rayleigh(scale=scale),
         'InvGamma': lambda shape, scale: stats.invgamma(shape, scale=scale),
         'Fixed': Fixed,
         'Triangular': lambda lo, ce, up: stats.triang(c=(ce-lo)/(up-lo),
@@ -420,7 +421,7 @@ def find_prior_parameters(prior):
     threepars = ['Triangular']
     twopars = [
         'LogUniform', 'ModifiedLogUniform', 'Gaussian', 'Kumaraswamy',
-        'Cauchy', 'InvGamma'
+        'Cauchy', 'Rayleigh', 'InvGamma'
     ]
 
     if name in threepars + twopars: