Normalizing insertion model

Author: joshfactorial

neat/models/models.py

@@ -59,7 +59,9 @@ class InsertionModel(VariantModel):
     def __init__(self,
                  insert_len_model: dict[int: float, ...],
                  rng: Generator = None):
-        self.insert_len_model = insert_len_model
+        # normalize the values
+        tot = sum(insert_len_model.values())
+        self.insertion_len_model = {key: val / tot for key, val in insert_len_model.items()}
         self.rng = rng
 
     def get_insertion_length(self, size: int = None) -> int | list[int, ...]:
@@ -90,9 +92,9 @@ class DeletionModel(VariantModel):
     def __init__(self,
                  deletion_len_model: dict[int: float, ...],
                  rng: Generator = None):
-        max_value = max(deletion_len_model.values())
         # normalize the values
-        self.deletion_len_model = {key: val/max_value for key, val in deletion_len_model.items()}
+        tot = sum(deletion_len_model.values())
+        self.deletion_len_model = {key: val/tot for key, val in deletion_len_model.items()}
         self.rng = rng
 
     def get_deletion_length(self, size: int = None) -> int | list[int, ...]: