view setup.py @ 93:36c141f0f0bd default tip

add tensorflow dependency + console scripts
author Jeff Hammel <k0scist@gmail.com>
date Sun, 17 Dec 2017 14:31:35 -0800
parents a7d8583f7482
children
line wrap: on
line source

"""
setup packaging script for tvii
"""

import os

version = "0.0"
dependencies = ['h5py',
                'matplotlib',
                'numerics',
                'numpy',
                'Pillow',
                'scikit-learn',
                'scipy',
                'tensorflow']

# allow use of setuptools/distribute or distutils
kw = {}
try:
    from setuptools import setup
    kw['entry_points'] = """
    [console_scripts]
    centroid = nettwerk.centroid:main
    gaussian = nettwerk.dataset.gauss:main
    k-means = nettwerk.kmeans:main
    line = nettwerk.dataset.line:main
    linear-regression = nettwerk.linear_regression:main
    nettwerk-mnist = nettwerk.mnist:main
    random-circle = nettwerk.dataset.circle:main
    random-data = nettwerk.dataset.rand:main
    sigmoid = nettwerk.sigmoid:main
"""
    kw['install_requires'] = dependencies
except ImportError:
    from distutils.core import setup
    kw['requires'] = dependencies


try:
    here = os.path.dirname(os.path.abspath(__file__))
    description = open(os.path.join(here, 'README.txt')).read()
except IOError:
    description = ''


setup(name='tvii',
      version=version,
      description="TV2",
      long_description=description,
      classifiers=[], # Get strings from http://www.python.org/pypi?%3Aaction=list_classifiers
      author='Jeff Hammel',
      author_email='k0scist@gmail.com',
      url='http://k0s.org/hg/tvii',
      license='',
      packages=['tvii'],
      include_package_data=True,
      setup_requires=['pytest-runner'],
      tests_require=['tox', 'pytest'],
      zip_safe=False,
      **kw
      )