fix import

Author: xinrong-meng

python/pyspark/ml/tests/test_functions.py

@@ -18,6 +18,7 @@
 
 import numpy as np
 
+from pyspark.loose_version import LooseVersion
 from pyspark.ml.functions import predict_batch_udf
 from pyspark.sql.functions import array, struct, col
 from pyspark.sql.types import ArrayType, DoubleType, IntegerType, StructType, StructField, FloatType
@@ -29,8 +30,6 @@
     pyarrow_requirement_message,
 )
 
-from python.pyspark.loose_version import LooseVersion
-
 
 @unittest.skipIf(
     not have_pandas or not have_pyarrow,