From 0eba595f7a52d12ce66846ca5d51cf96667eb89e Mon Sep 17 00:00:00 2001
From: ashuaibi7 <ashuaibi@princeton.edu>
Date: Wed, 4 Dec 2024 11:34:22 -0500
Subject: [PATCH] ensure probability of gene in bmr_pmf is nonzero to avoid
 computing log(0) and encountering warning

---
 src/dialect/models/gene.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/dialect/models/gene.py b/src/dialect/models/gene.py
index 91e19ec..c4b11ac 100644
--- a/src/dialect/models/gene.py
+++ b/src/dialect/models/gene.py
@@ -56,7 +56,7 @@ def compute_log_likelihood(self, pi):
         log_likelihood = sum(
             np.log(self.bmr_pmf.get(c, 0) * (1 - pi) + self.bmr_pmf.get(c - 1, 0) * pi)
             for c in self.counts
-            if c in self.bmr_pmf
+            if c in self.bmr_pmf and self.bmr_pmf[c] > 0
         )
         return log_likelihood