UPD: change gamma value error message

scikit-learn-contrib · Dec 21, 2023 · e8c2ba4 · e8c2ba4
1 parent 4bf1da2
commit e8c2ba4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/mapie/tests/test_utils.py b/mapie/tests/test_utils.py
@@ -459,7 +459,10 @@ def test_valid_gamma(gamma: float) -> None:
 @pytest.mark.parametrize("gamma", [1.5, -0.1])
 def test_invalid_large_gamma(gamma: float) -> None:
     """Test a non-valid gamma parameter."""
-    with pytest.raises(ValueError, match="Gamma must be between 0 and 1."):
+    with pytest.raises(
+        ValueError,
+        match="Invalid gamma. Allowed values are between 0 and 1."
+    ):
         check_gamma(gamma)
 
 

diff --git a/mapie/utils.py b/mapie/utils.py
@@ -368,7 +368,7 @@ def check_gamma(
     gamma: float
 ) -> None:
     """
-    Check if gamma is between 0 and 1
+    Check if gamma is between 0 and 1.
 
     Parameters
     ----------
@@ -379,10 +379,9 @@ def check_gamma(
     ValueError
         If gamma is lower than 0 or higher than 1.
     """
-
     if (gamma < 0) or (gamma > 1):
         raise ValueError(
-            "Gamma must be between 0 and 1."
+            "Invalid gamma. Allowed values are between 0 and 1."
         )