bd57af3879
The current checking does version `1.x' is less than `1.4' this will fail if x has greater than 1 digit, since x > 4, however `1.x` < `1.4`
It fails in my system since I have version `1.10` :P
Author: MechCoder <manojkumarsivaraj334@gmail.com>
Closes #6579 from MechCoder/np_ver and squashes the following commits:
15430f8 [MechCoder] fix syntax error
893fb7e [MechCoder] remove equal to
e35f0d4 [MechCoder] minor
e89376c [MechCoder] Better checking
22703dd [MechCoder] [SPARK-8032] Make version checking for NumPy in MLlib more robust
(cherry picked from commit 452eb82dd7
)
Signed-off-by: Xiangrui Meng <meng@databricks.com>
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one or more
|
|
# contributor license agreements. See the NOTICE file distributed with
|
|
# this work for additional information regarding copyright ownership.
|
|
# The ASF licenses this file to You under the Apache License, Version 2.0
|
|
# (the "License"); you may not use this file except in compliance with
|
|
# the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
#
|
|
|
|
"""
|
|
Python bindings for MLlib.
|
|
"""
|
|
from __future__ import absolute_import
|
|
|
|
# MLlib currently needs NumPy 1.4+, so complain if lower
|
|
|
|
import numpy
|
|
|
|
ver = [int(x) for x in numpy.version.version.split('.')[:2]]
|
|
if ver < [1, 4]:
|
|
raise Exception("MLlib requires NumPy 1.4+")
|
|
|
|
__all__ = ['classification', 'clustering', 'feature', 'fpm', 'linalg', 'random',
|
|
'recommendation', 'regression', 'stat', 'tree', 'util']
|