From 31b3c1df54fb2d3cb0aac9acbc0a13c274c64cd2 Mon Sep 17 00:00:00 2001
From: Patrick Wendell <pwendell@gmail.com>
Date: Wed, 17 Jul 2013 17:43:15 -0700
Subject: Small improvement in python script

---
 ec2/spark_ec2.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ec2/spark_ec2.py b/ec2/spark_ec2.py
index 413fcf2695..d1db535e8c 100755
--- a/ec2/spark_ec2.py
+++ b/ec2/spark_ec2.py
@@ -158,7 +158,7 @@ def is_active(instance):
 def get_spark_shark_version(opts):
   spark_shark_map = {"0.7.2": "0.7.0"}
   version = opts.spark_version.replace("v", "")
-  if version not in ["0.7.2"]:
+  if version not in spark_shark_map:
     print >> stderr, "Don't know about Spark version: %s" % version
     sys.exit(1)
   return (version, spark_shark_map[version])
-- 
cgit v1.2.3