isclose in adjust test

Orange-OpenSource · Nov 29, 2024 · e902e58 · e902e58
1 parent b13d61c
commit e902e58
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -8,9 +8,6 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
-        with:
-          python-version: 3.9
-
       - run: pip install --upgrade hatch
       - run: hatch fmt --check
       - run: hatch test --cover
diff --git a/tests/probe/test_adjust.py b/tests/probe/test_adjust.py
@@ -6,6 +6,8 @@
 # see the "LICENSE.md" file for more details
 # or https://github.com/Orange-OpenSource/mislabeled/blob/master/LICENSE.md
 
+import math
+
 import numpy as np
 from sklearn.datasets import make_moons
 from sklearn.linear_model import LogisticRegression
@@ -23,4 +25,4 @@ def test_means_per_class_when_adjusted_are_equals():
     confidence = Confidence(Adjust(Probabilities()))
     c = confidence(logreg, X, y)
 
-    assert np.mean(c[y == 0]) == np.mean(c[y == 1])
+    assert math.isclose(np.mean(c[y == 0]), np.mean(c[y == 1]))