From e9edb6fe96f585683f389ead4b3b9e86e2511d50 Mon Sep 17 00:00:00 2001
From: Fokko Driesprong <fokko@apache.org>
Date: Tue, 1 Sep 2020 21:07:37 +0200
Subject: [PATCH] [SPARK-32770][PYSPARK] Add missing imports

---
 python/pyspark/ml/classification.py | 1 +
 python/pyspark/ml/fpm.py            | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/python/pyspark/ml/classification.py b/python/pyspark/ml/classification.py
index c8e15ca1568cc..999765ce5d084 100644
--- a/python/pyspark/ml/classification.py
+++ b/python/pyspark/ml/classification.py
@@ -17,6 +17,7 @@
 
 import operator
 import sys
+import uuid
 from multiprocessing.pool import ThreadPool
 
 from pyspark import since, keyword_only
diff --git a/python/pyspark/ml/fpm.py b/python/pyspark/ml/fpm.py
index 0e302d53d97e7..329a1c164cd33 100644
--- a/python/pyspark/ml/fpm.py
+++ b/python/pyspark/ml/fpm.py
@@ -15,6 +15,8 @@
 # limitations under the License.
 #
 
+import sys
+
 from pyspark import keyword_only, since
 from pyspark.rdd import ignore_unicode_prefix
 from pyspark.sql import DataFrame