From 8289b6e1dd5dbcaba10c39eeb7a4452c3609949b Mon Sep 17 00:00:00 2001
From: Gregory Ashton <gregory.ashton@ligo.org>
Date: Mon, 11 May 2020 10:02:41 +1000
Subject: [PATCH] Simplify things
---
bilby/core/result.py | 5 +----
1 file changed, 1 insertion(+), 4 deletions(-)
diff --git a/bilby/core/result.py b/bilby/core/result.py
index 11ec22a0e..0555b8248 100644
--- a/bilby/core/result.py
+++ b/bilby/core/result.py
@@ -168,10 +168,7 @@ def rejection_sample(posterior, weights):
"""
keep = weights > np.random.uniform(0, max(weights), weights.shape)
- if isinstance(posterior, np.ndarray):
- return posterior[keep]
- elif isinstance(posterior, pd.DataFrame):
- return posterior.iloc[keep]
+ return posterior[keep]
def reweight(result, label=None, new_likelihood=None, new_prior=None,
--
GitLab