from setuptools import setup
import os
import sys

_here = os.path.abspath(os.path.dirname(__file__))

if sys.version_info[0] < 3:
    with open(os.path.join(_here, 'README.rst')) as f:
        long_description = f.read()
else:
    with open(os.path.join(_here, 'README.rst'), encoding='utf-8') as f:
        long_description = f.read()

setup(
    name='hess',
    version='alpha',
    description=('Repo for Hessian Loss Surface Inference'),
    long_description=long_description,
    author='Salkey, Benton, Maddox',
    author_email='gwb260@nyu.edu',
    url='https://github.com/g-benton/hess_truck',
    license='MPL-2.0',
    packages=['hess'],
   install_requires=[
    'matplotlib==3.0.3',
    'setuptools==41.0.0',
    'scipy>=1.2.1',
    'torch>=1.0.1',
    'numpy==1.16.2',
    'gpytorch>=0.3.1',
    'scikit_learn>=0.20.3'
   ],
    include_package_data=True,
    classifiers=[
        'Development Status :: 0',
        'Intended Audience :: Science/Research',
        'Programming Language :: Python :: 3.7'],
)
