Update sklearn/datasets/tests/test_svmlight_format.py

glemaitre · rth · web-flow · commit c56caea68694 · 2019-02-26T11:11:53.000+01:00
Co-Authored-By: rth &lt;rth.yurchak@pm.me&gt;
diff --git a/sklearn/datasets/tests/test_svmlight_format.py b/sklearn/datasets/tests/test_svmlight_format.py
@@ -263,7 +263,7 @@ def test_dump():
                     assert_array_equal(X2.sorted_indices().indices, X2.indices)
 
                     X2_dense = X2.toarray()
-                    if sp.issparse(X_input):
+                    X_input_dense = X_input.toarray() if sp.issparse(X_input) else X_input
                         X_input_dense = X_input.toarray()
                     else:
                         X_input_dense = X_input
