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

3
4
5
6
7
8
9
10
11
setup_attrs = {
    'name': 'torchani',
    'description': 'PyTorch implementation of ANI',
    '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
12
13
    'use_scm_version': True,
    'setup_requires': ['setuptools_scm'],
14
    'install_requires': [
15
16
        'torch-nightly',
        'pytorch-ignite-nightly',
17
18
19
20
        'lark-parser',
        'h5py',
    ],
    'test_suite': 'nose.collector',
21
22
23
24
    'tests_require': [
        'nose',
        'tensorboardX',
        'tqdm',
Gao, Xiang's avatar
Gao, Xiang committed
25
        'ase',
Gao, Xiang's avatar
Gao, Xiang committed
26
        'coverage',
27
    ],
28
29
30
}

setup(**setup_attrs)