Skip to content
This repository was archived by the owner on Dec 6, 2023. It is now read-only.

Commit 178d48a

Browse files
committed
fix incredibly weird nnz behavior
1 parent f06f871 commit 178d48a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

lightning/impl/tests/test_primal_cd.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -178,11 +178,11 @@ def test_debiasing_l1l2():
178178
multiclass=False,
179179
debiasing=True,
180180
warm_debiasing=warm_debiasing,
181-
max_iter=20, C=0.01, random_state=0,
182-
verbose=True)
181+
max_iter=20, C=0.01, random_state=0)
183182
clf.fit(mult_csc, mult_target)
184183
assert_greater(clf.score(mult_csc, mult_target), 0.75)
185-
assert_equal(clf.n_nonzero(percentage=True), 0.08)
184+
nnz = clf.n_nonzero(percentage=True)
185+
assert_equal(nnz, 0.08)
186186

187187

188188
def test_debiasing_warm_start():

0 commit comments

Comments
 (0)