Relax assert for number of non-zero elements in test (#191)

* Update test_primal_cd.py * address review comments
scikit-learn-contrib · Jan 3, 2022 · bbdee34 · bbdee34
1 parent 729b45c
commit bbdee34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lightning/impl/tests/test_primal_cd.py b/lightning/impl/tests/test_primal_cd.py
@@ -172,7 +172,7 @@ def test_debiasing_l1l2():
                            max_iter=20, C=0.01, random_state=0)
         clf.fit(mult_csc, mult_target)
         assert clf.score(mult_csc, mult_target) > 0.75
-        assert clf.n_nonzero(percentage=True) == 0.08
+        assert 0.0 <= clf.n_nonzero(percentage=True) <= 0.1
 
 
 def test_debiasing_warm_start():