fix parsing of Rayleigh distribution (#76)

j-faria · Nov 15, 2022 · f79729c · f79729c
1 parent 62a6e70
commit f79729c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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: