forked from pyg-team/pytorch_geometric
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
79 lines (71 loc) · 1.5 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
from setuptools import find_packages, setup
__version__ = '2.0.5'
URL = 'https://github.com/pyg-team/pytorch_geometric'
install_requires = [
'tqdm',
'numpy',
'scipy',
'jinja2',
'requests',
'pyparsing',
'scikit-learn',
]
graphgym_requires = [
'yacs',
'hydra-core',
'protobuf<4.21',
'pytorch-lightning',
]
full_requires = graphgym_requires + [
'h5py',
'numba',
'sympy',
'pandas',
'captum',
'rdflib',
'trimesh',
'networkx',
'tabulate',
'matplotlib',
'scikit-image',
'pytorch-memlab',
'torchmetrics>=0.7',
]
benchmark_requires = [
'protobuf<4.21',
'wandb',
]
test_requires = [
'pytest',
'pytest-cov',
]
dev_requires = test_requires + [
'pre-commit',
]
setup(
name='torch_geometric',
version=__version__,
description='Graph Neural Network Library for PyTorch',
author='Matthias Fey',
author_email='[email protected]',
url=URL,
download_url=f'{URL}/archive/{__version__}.tar.gz',
keywords=[
'deep-learning',
'pytorch',
'geometric-deep-learning',
'graph-neural-networks',
'graph-convolutional-networks',
],
python_requires='>=3.7',
install_requires=install_requires,
extras_require={
'graphgym': graphgym_requires,
'full': full_requires,
'benchmark': benchmark_requires,
'test': test_requires,
'dev': dev_requires,
},
packages=find_packages(),
include_package_data=True,
)