diff options
author | Matei Zaharia <matei@databricks.com> | 2014-01-14 12:27:58 -0800 |
---|---|---|
committer | Matei Zaharia <matei@databricks.com> | 2014-01-14 12:27:58 -0800 |
commit | 5b3a3e28d7666e5e93f3b4fc9c27714c751af9a8 (patch) | |
tree | 7814a2001343a2be9534b122a4cd1b33bc9c65c5 /python/pyspark/mllib | |
parent | 938e4a0e16ecd8e362c99f02a309d750060771a8 (diff) | |
download | spark-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/pyspark/mllib')
-rw-r--r-- | python/pyspark/mllib/__init__.py | 10 |
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+") |