aboutsummaryrefslogtreecommitdiff
path: root/examples/src/main/python/transitive_closure.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/src/main/python/transitive_closure.py')
-rwxr-xr-xexamples/src/main/python/transitive_closure.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/examples/src/main/python/transitive_closure.py b/examples/src/main/python/transitive_closure.py
index 744cce6651..8698369b13 100755
--- a/examples/src/main/python/transitive_closure.py
+++ b/examples/src/main/python/transitive_closure.py
@@ -36,11 +36,11 @@ def generateGraph():
if __name__ == "__main__":
- if len(sys.argv) == 1:
- print >> sys.stderr, "Usage: transitive_closure <master> [<slices>]"
- exit(-1)
- sc = SparkContext(sys.argv[1], "PythonTransitiveClosure")
- slices = int(sys.argv[2]) if len(sys.argv) > 2 else 2
+ """
+ Usage: transitive_closure [slices]
+ """
+ sc = SparkContext(appName="PythonTransitiveClosure")
+ slices = int(sys.argv[1]) if len(sys.argv) > 1 else 2
tc = sc.parallelize(generateGraph(), slices).cache()
# Linear transitive closure: each round grows paths by one edge,