diff --git a/.gitignore b/.gitignore index cac42814dd0a1cf3988e6ccdbe38e5400181155d..88717818866b233ab49b644f24d116aa61efea91 100644 --- a/.gitignore +++ b/.gitignore @@ -14,4 +14,4 @@ MANIFEST *.version *.ipynb_checkpoints outdir/* -.idea/* \ No newline at end of file +.idea/* diff --git a/bilby/core/prior/analytical.py b/bilby/core/prior/analytical.py index 4de79a382021dbc679e3eb81a7035b16f91fcb28..9b1d52898fcd9bdb190636f96e9151c54c4a2a1b 100644 --- a/bilby/core/prior/analytical.py +++ b/bilby/core/prior/analytical.py @@ -991,15 +991,15 @@ class Beta(Prior): - betaln(self.alpha, self.beta) - xlogy(self.alpha + self.beta - 1, self.maximum - self.minimum) # deal with the fact that if alpha or beta are < 1 you get infinities at 0 and 1 - if isinstance(val, np.ndarray): + if isinstance(val, (float, int)): + if np.isfinite(_ln_prob) and self.minimum <= val <= self.maximum: + return _ln_prob + return -np.inf + else: _ln_prob_sub = -np.inf * np.ones(len(val)) idx = np.isfinite(_ln_prob) & (val >= self.minimum) & (val <= self.maximum) _ln_prob_sub[idx] = _ln_prob[idx] return _ln_prob_sub - else: - if np.isfinite(_ln_prob) and val >= self.minimum and val <= self.maximum: - return _ln_prob - return -np.inf def cdf(self, val): if isinstance(val, (float, int)):