setup.py

#!/usr/bin/env python

from setuptools import setup

import splearn
import sys


def is_numpy_installed():
    try:
        import numpy
    except ImportError:
        return False
    return True


def setup_package():
    metadata = dict(
        name='sparkit-learn',
        version=str(splearn.__version__),
        description='Scikit-learn on PySpark',
        author='Krisztian Szucs, Andras Fulop',
        author_email='krisztian.szucs@lensa.com, andras.fulop@lensa.com',
        license='Apache License, Version 2.0',
        url='https://github.com/lensacom/sparkit-learn',
        packages=['splearn',
                  'splearn.rdd',
                  'splearn.cluster',
                  'splearn.decomposition',
                  'splearn.feature_extraction',
                  'splearn.feature_selection',
                  'splearn.linear_model',
                  'splearn.svm'],
        
        long_description=open('./README.rst').read(),
        install_requires=open('./requirements.txt').read().split()
    )
    if not (len(sys.argv) >= 2
            and ('--help' in sys.argv[1:] or sys.argv[1]
                 in ('--help-commands', 'egg_info', '--version', 'clean'))):
        if is_numpy_installed() is False:
            raise ImportError("Numerical Python (NumPy) is not installed.\n"
                              "sparkit-learn requires NumPy.\n"
                              "Installation instructions are available on scikit-learn website: "
                              "http://scikit-learn.org/stable/install.html\n")

    setup(**metadata)


if __name__ == '__main__':
    setup_package()