From 7272e630311553c169e0570d6d038937da9d506e Mon Sep 17 00:00:00 2001 From: Max Shinn Date: Wed, 1 Dec 2021 15:05:51 +0000 Subject: [PATCH] Get rid of needless renormalization error, fixes #43 --- ddm/solution.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ddm/solution.py b/ddm/solution.py index 4a3b73b..068be2b 100644 --- a/ddm/solution.py +++ b/ddm/solution.py @@ -317,8 +317,8 @@ def resample(self, k=1, seed=0): shift = np.max(shorter_t_domain)+1 combined_domain = list(shorter_t_domain) + list(shorter_t_domain+shift) + [-1] combined_probs = list(shorter_pdf_corr*self.model.dt) + list(shorter_pdf_err*self.model.dt) + [self.prob_undecided()] - if fsum(combined_probs) != 1: - print("Warning, distribution sums to %f rather than 1" % fsum(combined_probs)) + if np.abs(np.sum(combined_probs)-1) >= .0001: + print("Warning, distribution sums to %f rather than 1" % np.sum(combined_probs)) samp = np.random.choice(combined_domain, p=combined_probs, replace=True, size=k) undecided = np.sum(samp==-1) samp = samp[samp != -1] # Remove undecided trials