From c1690a089217c2c168afe795d719222de3490d84 Mon Sep 17 00:00:00 2001
From: Colm Talbot <colm.talbot@ligo.org>
Date: Sun, 31 Jan 2021 20:29:28 -0600
Subject: [PATCH] Prior normalization cache

---
 bilby/core/prior/dict.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/bilby/core/prior/dict.py b/bilby/core/prior/dict.py
index 3ec1a0593..f79c62246 100644
--- a/bilby/core/prior/dict.py
+++ b/bilby/core/prior/dict.py
@@ -391,6 +391,7 @@ class PriorDict(dict):
             samples = self.sample_subset(keys=keys, size=sampling_chunk)
             keep = np.atleast_1d(self.evaluate_constraints(samples))
             if len(keep) == 1:
+                self._cached_normalizations[keys] = 1
                 return 1
             all_samples = {key: np.array([]) for key in keys}
             while np.count_nonzero(keep) < min_accept:
-- 
GitLab