scikit-learn
diff --git a/‎sklearn/linear_model/tests/test_sgd.py
Lines changed: 5 additions & 5 deletions b/‎sklearn/linear_model/tests/test_sgd.py
Lines changed: 5 additions & 5 deletions
@@ -1394,7 +1394,7 @@ def asgd_oneclass(klass, X, eta, nu, coef_init=None, offset_init=0.0):
     decay = 1.0
 
     # sparse data has a fixed decay of .01
-    if klass in (SparseSGDOneClassSVM):
+    if klass == SparseSGDOneClassSVM:
         decay = .01
 
     for i, entry in enumerate(X):
@@ -1542,7 +1542,7 @@ def test_late_onset_averaging_reached_oneclass(klass):
     # Start from clf2 solution, compute averaging using asgd function and
     # compare with clf1 solution
     average_coef, average_offset = \
-        asgd_oneclass(X, eta0, nu,
+        asgd_oneclass(klass, X, eta0, nu,
                       coef_init=clf2.coef_.ravel(),
                       offset_init=clf2.offset_)
 
@@ -1568,7 +1568,7 @@ def test_sgd_averaged_computed_correctly_oneclass(klass):
                 max_iter=1, average=True, shuffle=False)
 
     clf.fit(X)
-    average_coef, average_offset = asgd_oneclass(X, eta, nu)
+    average_coef, average_offset = asgd_oneclass(klass, X, eta, nu)
 
     assert_array_almost_equal(clf.coef_, average_coef, decimal=16)
     assert_almost_equal(clf.offset_, average_offset, decimal=15)
@@ -1591,7 +1591,7 @@ def test_sgd_averaged_partial_fit_oneclass(klass):
 
     clf.partial_fit(X[:int(n_samples / 2)][:])
     clf.partial_fit(X[int(n_samples / 2):][:])
-    average_coef, average_offset = asgd_oneclass(X, eta, nu)
+    average_coef, average_offset = asgd_oneclass(klass, X, eta, nu)
 
     assert_array_almost_equal(clf.coef_, average_coef, decimal=16)
     assert_almost_equal(clf.offset_, average_offset, decimal=15)
@@ -1611,7 +1611,7 @@ def test_average_sparse_oneclass(klass):
 
     clf.partial_fit(X3[:int(n_samples / 2)])
     clf.partial_fit(X3[int(n_samples / 2):])
-    average_coef, average_offset = asgd_oneclass(X3, eta, nu)
+    average_coef, average_offset = asgd_oneclass(klass, X3, eta, nu)
 
     assert_array_almost_equal(clf.coef_, average_coef, decimal=16)
     assert_almost_equal(clf.offset_, average_offset, decimal=15)