diff --git a/python/pyspark/testing/utils.py b/python/pyspark/testing/utils.py index fe25136864eef..8a7aa405e4ac7 100644 --- a/python/pyspark/testing/utils.py +++ b/python/pyspark/testing/utils.py @@ -38,7 +38,7 @@ have_scipy = False have_numpy = False try: - import scipy.sparse # noqa: F401 + import scipy # noqa: F401 have_scipy = True except ImportError: