aboutsummaryrefslogtreecommitdiff
path: root/examples/src/main/python/pi.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/src/main/python/pi.py')
-rwxr-xr-xexamples/src/main/python/pi.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/examples/src/main/python/pi.py b/examples/src/main/python/pi.py
index ab0645fc2f..234720b55f 100755
--- a/examples/src/main/python/pi.py
+++ b/examples/src/main/python/pi.py
@@ -23,11 +23,11 @@ from pyspark import SparkContext
if __name__ == "__main__":
- if len(sys.argv) == 1:
- print >> sys.stderr, "Usage: pi <master> [<slices>]"
- exit(-1)
- sc = SparkContext(sys.argv[1], "PythonPi")
- slices = int(sys.argv[2]) if len(sys.argv) > 2 else 2
+ """
+ Usage: pi [slices]
+ """
+ sc = SparkContext(appName="PythonPi")
+ slices = int(sys.argv[1]) if len(sys.argv) > 1 else 2
n = 100000 * slices
def f(_):
x = random() * 2 - 1