aboutsummaryrefslogtreecommitdiff
path: root/python
diff options
context:
space:
mode:
authorMatei Zaharia <matei@databricks.com>2014-01-14 12:27:58 -0800
committerMatei Zaharia <matei@databricks.com>2014-01-14 12:27:58 -0800
commit5b3a3e28d7666e5e93f3b4fc9c27714c751af9a8 (patch)
tree7814a2001343a2be9534b122a4cd1b33bc9c65c5 /python
parent938e4a0e16ecd8e362c99f02a309d750060771a8 (diff)
downloadspark-5b3a3e28d7666e5e93f3b4fc9c27714c751af9a8.tar.gz
spark-5b3a3e28d7666e5e93f3b4fc9c27714c751af9a8.tar.bz2
spark-5b3a3e28d7666e5e93f3b4fc9c27714c751af9a8.zip
Complain if Python and NumPy versions are too old for MLlib
Diffstat (limited to 'python')
-rw-r--r--python/pyspark/mllib/__init__.py10
1 files changed, 10 insertions, 0 deletions
diff --git a/python/pyspark/mllib/__init__.py b/python/pyspark/mllib/__init__.py
index b1a5df109b..b420d7a7f2 100644
--- a/python/pyspark/mllib/__init__.py
+++ b/python/pyspark/mllib/__init__.py
@@ -18,3 +18,13 @@
"""
Python bindings for MLlib.
"""
+
+# MLlib currently needs Python 2.7+ and NumPy 1.7+, so complain if lower
+
+import sys
+if sys.version_info[0:2] < (2, 7):
+ raise Exception("MLlib requires Python 2.7+")
+
+import numpy
+if numpy.version.version < '1.7':
+ raise Exception("MLlib requires NumPy 1.7+")