setup.py

import os

from setuptools import setup, find_packages

with open('cw_nets/__init__.py') as f:
    for line in f:
        if line.find("__version__") >= 0:
            version = line.split("=")[1].strip()
            version = version.strip('"')
            version = version.strip("'")
            continue


with open('README.rst') as f:
    readme = f.read()

# Runtime requirements.
inst_reqs = ["cw-tiler", 
             "numpy",
             "tqdm",
             "shapely",
             "rasterio",
             "opencv-contrib-python-headless",
             "scikit-image",
             "tensorflow-gpu", "keras", "scikit-learn", "torch", "torchvision"]

extra_reqs = {
    'test': ['mock', 'pytest', 'pytest-cov', 'codecov'],
    'gpu' : ['tensorflow-gpu'],
    'cpu' : ['tensorflow']
}
        

setup(name='cw_nets',
      version=version,
      description=u"""Run basic Segmentation on  Dataset or arbitrary GeoTIffs""",
      long_description=readme,
      classifiers=[
          'Intended Audience :: Information Technology',
          'Intended Audience :: Science/Research',
          'License :: OSI Approved :: Apache License',
          'Programming Language :: Python :: 3.6',
          'Topic :: Scientific/Engineering :: GIS'],
      keywords='segmentation spacenet machinelearning',
      scripts=['cw_nets/scripts/create_mask.py'],
      author=u"David Lindenbaum",
      author_email='dlindenbaum@iqt.org',
      url='https://github.com/CosmiQ/cw-nets',
      license='Apache 2.0',
      packages=find_packages(exclude=['ez_setup', 'examples', 'tests']),
      zip_safe=False,
      install_requires=inst_reqs,
      extras_require=extra_reqs)