added LinearOperator support in safe_sparse_dot()

doc/whats_new/v0.23.rst

@@ -341,7 +341,8 @@ Changelog
   pandas sparse DataFrame.
   :pr:`16021` by :user:`Rushabh Vasani <rushabh-v>`.
 
-- |Feature| :func:`utils.extmath.randomized_svd` now accepts 
+- |Feature| :func:`utils.extmath.randomized_svd` and 
+  :func:`utils.extmath.safe_dot_product` now accepts 
   :class:`scipy.sparse.linalg.LinearOperator` as argument. 
   :pr:`#16463` by :user:`Pavel V. Stishenko <Mazay0>`.
 

sklearn/utils/tests/test_extmath.py

@@ -7,7 +7,7 @@
 import numpy as np
 from scipy import sparse
 from scipy import linalg
-from scipy.sparse.linalg import aslinearoperator
+from scipy.sparse.linalg import LinearOperator
 from scipy import stats
 from scipy.special import expit
 
@@ -730,14 +730,15 @@ def test_safe_sparse_dot_operator():
     A = rng.random_sample((10, 20))
     B = rng.random_sample((20, 30))
     expected = np.dot(A, B)
-    A = aslinearoperator(A)
-    actual = safe_sparse_dot(A, B)
+    op = LinearOperator(A.shape, matvec=lambda x: np.dot(A, x))
+    actual = safe_sparse_dot(op, B)
     assert_allclose(actual, expected)
 
     # ndarray @ LinearOperator
     A = rng.random_sample((10, 20))
     B = rng.random_sample((20, 30))
     expected = np.dot(A, B)
-    B = aslinearoperator(B)
-    actual = safe_sparse_dot(A, B)
+    op = LinearOperator(B.shape, matvec=lambda x: np.dot(B, x),
+                        rmatvec=lambda x: np.dot(B.T, x))
+    actual = safe_sparse_dot(A, op)
     assert_allclose(actual, expected)