diff --git a/python/pyspark/mllib/tests.py b/python/pyspark/mllib/tests.py index 5af7d74999c06..886dbcf9aa982 100644 --- a/python/pyspark/mllib/tests.py +++ b/python/pyspark/mllib/tests.py @@ -24,6 +24,7 @@ import tempfile import array as pyarray from time import time, sleep + from numpy import array, array_equal, zeros, inf, all, random from numpy import sum as array_sum from py4j.protocol import Py4JJavaError