aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/streaming/flume.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyspark/streaming/flume.py')
-rw-r--r--python/pyspark/streaming/flume.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/python/pyspark/streaming/flume.py b/python/pyspark/streaming/flume.py
index c0cdc50d8d..b3d1905365 100644
--- a/python/pyspark/streaming/flume.py
+++ b/python/pyspark/streaming/flume.py
@@ -20,7 +20,7 @@ if sys.version >= "3":
from io import BytesIO
else:
from StringIO import StringIO
-from py4j.java_gateway import Py4JJavaError
+from py4j.protocol import Py4JJavaError
from pyspark.storagelevel import StorageLevel
from pyspark.serializers import PairDeserializer, NoOpSerializer, UTF8Deserializer, read_int