Fix the issue that StringIO doesn't work in Python 3

apache · Jun 20, 2015 · 152364c · 152364c
1 parent 14ba0ff
commit 152364c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/pyspark/streaming/flume.py b/python/pyspark/streaming/flume.py
@@ -15,7 +15,10 @@
 # limitations under the License.
 #
 
-from StringIO import StringIO
+if sys.version >= "3":
+    from io import StringIO
+else:
+    from StringIO import StringIO
 from py4j.java_gateway import Py4JJavaError
 
 from pyspark.storagelevel import StorageLevel