aboutsummaryrefslogtreecommitdiff
path: root/pyspark/examples/wordcount.py
diff options
context:
space:
mode:
Diffstat (limited to 'pyspark/examples/wordcount.py')
-rw-r--r--pyspark/examples/wordcount.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/pyspark/examples/wordcount.py b/pyspark/examples/wordcount.py
index 8365c070e8..857160624b 100644
--- a/pyspark/examples/wordcount.py
+++ b/pyspark/examples/wordcount.py
@@ -1,6 +1,8 @@
import sys
from operator import add
-from pyspark.context import SparkContext
+
+from pyspark import SparkContext
+
if __name__ == "__main__":
if len(sys.argv) < 3: