setup.py 775 Bytes
Newer Older
1
from setuptools import setup, find_packages
2
import sys
Xiang Gao's avatar
Xiang Gao committed
3

4
5
6
with open("README.md", "r") as fh:
    long_description = fh.read()

7
8
9
setup_attrs = {
    'name': 'torchani',
    'description': 'PyTorch implementation of ANI',
10
11
    'long_description': long_description,
    'long_description_content_type': "text/markdown",
12
13
14
15
16
17
    'url': 'https://github.com/zasdfgbnm/torchani',
    'author': 'Xiang Gao',
    'author_email': 'qasdfgtyuiop@ufl.edu',
    'license': 'MIT',
    'packages': find_packages(),
    'include_package_data': True,
Gao, Xiang's avatar
Gao, Xiang committed
18
19
    'use_scm_version': True,
    'setup_requires': ['setuptools_scm'],
20
    'install_requires': [
21
        'torch',
22
23
24
25
        'lark-parser',
    ],
}

26
27
28
if sys.version_info[0] < 3:
    setup_attrs['install_requires'].append('typing')

29
setup(**setup_attrs)