fix 2 warnings in test due to np.matrix being used (#214)

mathurinm · Dec 10, 2021 · f7fa260 · f7fa260
1 parent 69f27f2
commit f7fa260
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/celer/tests/test_lasso.py b/celer/tests/test_lasso.py
@@ -198,7 +198,7 @@ def test_weights():
         **params)
 
     alphas2, coefs2, gaps2 = celer_path(
-        X / weights[None, :], y, "lasso", **params)
+        X.multiply(1 / weights[None, :]), y, "lasso", **params)
 
     assert_allclose(alphas1, alphas2)
     assert_allclose(
@@ -208,7 +208,8 @@ def test_weights():
 
     alpha = 0.001
     clf1 = Lasso(alpha=alpha, weights=weights, fit_intercept=False).fit(X, y)
-    clf2 = Lasso(alpha=alpha, fit_intercept=False).fit(X / weights, y)
+    clf2 = Lasso(alpha=alpha, fit_intercept=False).fit(
+        X.multiply(1. / weights), y)
 
     assert_allclose(clf1.coef_, clf2.coef_ / weights)