fix

Author: xadupre

_unittests/ut_mlmodel/test_piecewise_decision_tree_experiment_fast.py

@@ -136,7 +136,7 @@ def test_criterions(self):
             _test_criterion_update(c2, i)
             left1, right1 = _test_criterion_node_impurity_children(c1)
             left2, right2 = _test_criterion_node_impurity_children(c2)
-            self.assertAlmostEqual(left1, left2)
+            self.assertAlmostEqual(left1, left2, atol=1e-8)
             self.assertAlmostEqual(right1, right2)
             v1 = _test_criterion_node_value(c1)
             v2 = _test_criterion_node_value(c2)

requirements-dev.txt

@@ -31,7 +31,7 @@ sphinx-issues; sys_platform == 'linux'
 sphinx-runpython
 toml; python_version < '3.11'
 tomli
-torch>=2.9.0
+torch>=2.9.0; sys_platform != 'darwin'
 torchvision
 torchaudio
 tqdm