mac

Author: xadupre

_unittests/ut_mlmodel/test_piecewise_decision_tree_experiment_fast.py

@@ -102,15 +102,15 @@ def test_criterions(self):
             assert_criterion_equal(c1, c2)
             left1, right1 = _test_criterion_node_impurity_children(c1)
             left2, right2 = _test_criterion_node_impurity_children(c2)
-            self.assertAlmostEqual(left1, left2)
-            self.assertAlmostEqual(right1, right2)
+            self.assertAlmostEqual(left1, left2, atol=1e-8)
+            self.assertAlmostEqual(right1, right2, atol=1e-8)
             v1 = _test_criterion_node_value(c1)
             v2 = _test_criterion_node_value(c2)
             self.assertEqual(v1, v2)
             assert_criterion_equal(c1, c2)
             p1 = _test_criterion_impurity_improvement(c1, 0.0, left1, right1)
             p2 = _test_criterion_impurity_improvement(c2, 0.0, left2, right2)
-            self.assertAlmostEqual(p1, p2)
+            self.assertAlmostEqual(p1, p2, atol=1e-8)
 
         X = numpy.array([[1.0, 2.0, 10.0, 11.0]]).T
         y = numpy.array([0.9, 1.1, 1.9, 2.1])
@@ -123,7 +123,7 @@ def test_criterions(self):
         _test_criterion_init(c2, ys, w, 1.0, ind, 1, y.shape[0])
         i1 = _test_criterion_node_impurity(c1)
         i2 = _test_criterion_node_impurity(c2)
-        self.assertAlmostEqual(i1, i2)
+        self.assertAlmostEqual(i1, i2, atol=1e-8)
         v1 = _test_criterion_node_value(c1)
         v2 = _test_criterion_node_value(c2)
         self.assertEqual(v1, v2)

pyproject.toml

@@ -118,7 +118,7 @@ repair-wheel-command = "auditwheel-symbols --manylinux 2_28 {wheel} ; abi3audit
 [tool.cibuildwheel.macos]
 archs = "arm64"  # or "universal2" for a single universal wheel
 environment = """
-    MACOSX_DEPLOYMENT_TARGET=14.0
+    MACOSX_DEPLOYMENT_TARGET=15.0
     LDFLAGS='-L$(brew --prefix libomp)/lib'
     CPPFLAGS='-I$(brew --prefix libomp)/include'
     CFLAGS='-I$(brew --prefix libomp)/include -arch x86_64 -arch arm64'

requirements-dev.txt

@@ -30,7 +30,7 @@ sphinx-issues; sys_platform == 'linux'
 sphinx-runpython
 toml; python_version < '3.11'
 tomli
-torch
+torch==2.9.1+cpu
 torchvision
 torchaudio
 tqdm