From d90badc02594cbdd49fe3b55b8325828424deaf9 Mon Sep 17 00:00:00 2001
From: szhan <shing.zhan@gmail.com>
Date: Tue, 2 Jul 2024 17:31:49 +0100
Subject: [PATCH] Check mu

---
 lshmm/core.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lshmm/core.py b/lshmm/core.py
index 7978a10..0b41ae2 100644
--- a/lshmm/core.py
+++ b/lshmm/core.py
@@ -313,6 +313,8 @@ def get_emission_matrix_haploid_hkylike(mu, kappa=None):
     :param float mu: Probability of mutation to any allele.
     :param float kappa: Transition-to-transversion rate ratio.
     """
+    if mu < 0.0 or mu > 1.0:
+        raise ValueError("Probability of mutation must be in [0, 1].")
     if kappa is not None:
         if kappa <= 0:
             raise ValueError("Transition-to-transversion rate ratio must be positive.")