setup.py

"""A setuptools based setup module.

See:
https://packaging.python.org/en/latest/distributing.html
https://github.com/pypa/sampleproject
"""

# Always prefer setuptools over distutils
from setuptools import setup, find_packages, Extension
# To use a consistent encoding
from codecs import open
from os import path
import os

here = path.abspath(path.dirname(__file__))

# Get the long description from the relevant file
with open(path.join(here, 'README.md'), encoding='utf-8') as f:
    long_description = f.read()

# C library for the GFL solver
os.environ["CC"] = "gcc"
os.environ["CXX"] = "g++"
module1 = Extension('libgraphfl',
                    include_dirs = ['cpp/include/'],
                    libraries=['gsl', 'gslcblas'],
                    sources = ['cpp/src/bayes_gfl.c', 'cpp/src/csparse.c', 'cpp/src/graph_fl.c', 'cpp/src/graph_tf.c', 'cpp/src/polyagamma.c', 'cpp/src/tf_dp.c', 'cpp/src/utils.c'])

setup(
    name='pygfl',
    version='1.0.2',
    description='A Fast and Flexible Graph-Fused Lasso Solver',
    long_description=long_description,
    url='https://github.com/tansey/gfl',
    author='Wesley Tansey',
    author_email='wes.tansey@gmail.com',
    license='LGPL',
    classifiers=[
        'Development Status :: 3 - Alpha',
        'Intended Audience :: Science/Research',
        'Topic :: Scientific/Engineering',
        'License :: OSI Approved :: GNU Lesser General Public License v3 or later (LGPLv3+)',
        'Programming Language :: Python :: 2',
        'Programming Language :: Python :: 2.7'
    ],
    keywords='statistics machinelearning lasso fusedlasso',
    packages=find_packages(exclude=['contrib', 'docs', 'tests*']),
    install_requires=['numpy', 'scipy', 'networkx'],
    package_data={
        'pygfl': [],
    },
    entry_points={
        'console_scripts': [
            'trails=pygfl.trails:main',
            'graphfl=pygfl:main',
            'imtv=pygfl:imtv'
        ],
    },
    ext_modules=[module1]
)