From 3d28bbf9f218ce50ab08fb3e9e62ed9e2fc2307b Mon Sep 17 00:00:00 2001
From: Benjamin Peterson <benjamin@python.org>
Date: Wed, 28 Feb 2018 08:56:55 -0800
Subject: [PATCH] add missing sys imports

---
 python/pyspark/ml/clustering.py        | 2 ++
 python/pyspark/ml/recommendation.py    | 2 ++
 python/pyspark/ml/regression.py        | 1 +
 python/pyspark/mllib/classification.py | 1 +
 python/pyspark/mllib/evaluation.py     | 1 +
 python/pyspark/mllib/tree.py           | 1 +
 6 files changed, 8 insertions(+)

diff --git a/python/pyspark/ml/clustering.py b/python/pyspark/ml/clustering.py
index cf4294caf4b02..b3d5fb17f6b81 100644
--- a/python/pyspark/ml/clustering.py
+++ b/python/pyspark/ml/clustering.py
@@ -15,6 +15,8 @@
 # limitations under the License.
 #
 
+import sys
+
 from pyspark import since, keyword_only
 from pyspark.ml.util import *
 from pyspark.ml.wrapper import JavaEstimator, JavaModel, JavaWrapper
diff --git a/python/pyspark/ml/recommendation.py b/python/pyspark/ml/recommendation.py
index 3bc6f82f7596e..a8eae9bd268d3 100644
--- a/python/pyspark/ml/recommendation.py
+++ b/python/pyspark/ml/recommendation.py
@@ -15,6 +15,8 @@
 # limitations under the License.
 #
 
+import sys
+
 from pyspark import since, keyword_only
 from pyspark.ml.util import *
 from pyspark.ml.wrapper import JavaEstimator, JavaModel
diff --git a/python/pyspark/ml/regression.py b/python/pyspark/ml/regression.py
index a44c9fbad3a8e..de0a0fa9f3bf8 100644
--- a/python/pyspark/ml/regression.py
+++ b/python/pyspark/ml/regression.py
@@ -15,6 +15,7 @@
 # limitations under the License.
 #
 
+import sys
 import warnings
 
 from pyspark import since, keyword_only
diff --git a/python/pyspark/mllib/classification.py b/python/pyspark/mllib/classification.py
index 59a052e1217f3..bb281981fd56b 100644
--- a/python/pyspark/mllib/classification.py
+++ b/python/pyspark/mllib/classification.py
@@ -16,6 +16,7 @@
 #
 
 from math import exp
+import sys
 import warnings
 
 import numpy
diff --git a/python/pyspark/mllib/evaluation.py b/python/pyspark/mllib/evaluation.py
index 4392579b847bf..36cb03369b8c0 100644
--- a/python/pyspark/mllib/evaluation.py
+++ b/python/pyspark/mllib/evaluation.py
@@ -15,6 +15,7 @@
 # limitations under the License.
 #
 
+import sys
 import warnings
 
 from pyspark import since
diff --git a/python/pyspark/mllib/tree.py b/python/pyspark/mllib/tree.py
index 20f899e2c24cd..b05734ce489d9 100644
--- a/python/pyspark/mllib/tree.py
+++ b/python/pyspark/mllib/tree.py
@@ -17,6 +17,7 @@
 
 from __future__ import absolute_import
 
+import sys
 import random
 
 from pyspark import SparkContext, RDD, since