aboutsummaryrefslogtreecommitdiff
path: root/examples/src/main/python/avro_inputformat.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/src/main/python/avro_inputformat.py')
-rw-r--r--examples/src/main/python/avro_inputformat.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/examples/src/main/python/avro_inputformat.py b/examples/src/main/python/avro_inputformat.py
index 4626bbb7e3..da368ac628 100644
--- a/examples/src/main/python/avro_inputformat.py
+++ b/examples/src/main/python/avro_inputformat.py
@@ -15,9 +15,12 @@
# limitations under the License.
#
+from __future__ import print_function
+
import sys
from pyspark import SparkContext
+from functools import reduce
"""
Read data file users.avro in local Spark distro:
@@ -49,7 +52,7 @@ $ ./bin/spark-submit --driver-class-path /path/to/example/jar \
"""
if __name__ == "__main__":
if len(sys.argv) != 2 and len(sys.argv) != 3:
- print >> sys.stderr, """
+ print("""
Usage: avro_inputformat <data_file> [reader_schema_file]
Run with example jar:
@@ -57,7 +60,7 @@ if __name__ == "__main__":
/path/to/examples/avro_inputformat.py <data_file> [reader_schema_file]
Assumes you have Avro data stored in <data_file>. Reader schema can be optionally specified
in [reader_schema_file].
- """
+ """, file=sys.stderr)
exit(-1)
path = sys.argv[1]
@@ -77,6 +80,6 @@ if __name__ == "__main__":
conf=conf)
output = avro_rdd.map(lambda x: x[0]).collect()
for k in output:
- print k
+ print(k)
sc.stop()