black flake

LLNL · Oct 16, 2024 · 701759e · 701759e
1 parent 88e6368
commit 701759e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/thicket/stats/confidence_interval.py b/thicket/stats/confidence_interval.py
@@ -29,7 +29,7 @@ def confidence_interval(thicket, columns=None, confidence_level=0.95):
         columns (list): List of hardware/timing metrics to perform confidence interval
             calculation on. Note, if using a columnar_joined thicket a list of tuples
             must be passed in with the format (column index, column name).
-        confidence_level (int,float):  The confidence level (often 0.90, 0.95, or 0.99)
+        confidence_level (float):  The confidence level (often 0.90, 0.95, or 0.99)
             indicates the degree of confidence that the true parameter lies within the interval.
 
     Returns:
@@ -44,9 +44,7 @@ def confidence_interval(thicket, columns=None, confidence_level=0.95):
         raise ValueError("Value passed to 'columns' must be of type list.")
 
     if not isinstance(confidence_level, float):
-        raise ValueError(
-            r"Value passed to 'confidence_level' must be of type float or int."
-        )
+        raise ValueError(r"Value passed to 'confidence_level' must be of type float.")
 
     if confidence_level >= 1 or confidence_level <= 0:
         raise ValueError(

diff --git a/thicket/tests/test_stats.py b/thicket/tests/test_stats.py
@@ -1233,7 +1233,7 @@ def test_confidence_interval(thicket_axis_columns):
 
     with pytest.raises(
         ValueError,
-        match="Value passed to 'confidence_level' must be of type float or int.",
+        match="Value passed to 'confidence_level' must be of type float.",
     ):
         th.stats.confidence_interval(
             combined_th, columns=columns, confidence_level="0.95"
@@ -1243,7 +1243,7 @@ def test_confidence_interval(thicket_axis_columns):
         ValueError,
         match=r"Value passed to 'confidence_level' must be in the range of \(0, 1\).",
     ):
-        th.stats.confidence_interval(combined_th, columns=columns, confidence_level=95)
+        th.stats.confidence_interval(combined_th, columns=columns, confidence_level=1.2)
 
     # Hardcoded cases
     columns = [("block_128", "Avg time/rank"), ("default", "Avg time/rank")]