Merge pull request #637 from broadinstitute/konradjk-patch-1

Update constraint.py
broadinstitute · Oct 25, 2023 · dc36048 · dc36048
2 parents f0b8c42 + b51d3c4
commit dc36048
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gnomad/utils/constraint.py b/gnomad/utils/constraint.py
@@ -1095,8 +1095,8 @@ def compute_pli(
         expected_values = {"Null": 1.0, "Rec": 0.463, "LI": 0.089}
 
     # Set up initial values.
-    last_pi = {"Null": 0, "Rec": 0, "LI": 0}
-    pi = {"Null": 1 / 3, "Rec": 1 / 3, "LI": 1 / 3}
+    last_pi = {k: 0 for k in expected_values.keys()}
+    pi = {k: 1 / len(expected_values.keys()) for k in expected_values.keys()}
 
     dpois_expr = {
         k: hl.or_missing(