Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
34 lines (30 sloc) 883 Bytes
from setuptools import setup, find_packages
__version__ = '1.1.3'
url = 'https://github.com/rusty1s/pytorch_geometric'
install_requires = [
'numpy',
'scipy',
'networkx',
'scikit-learn',
'plyfile',
'pandas',
'rdflib',
]
setup_requires = ['pytest-runner']
tests_require = ['pytest', 'pytest-cov', 'mock']
setup(
name='torch_geometric',
version=__version__,
description='Geometric Deep Learning Extension Library for PyTorch',
author='Matthias Fey',
author_email='matthias.fey@tu-dortmund.de',
url=url,
download_url='{}/archive/{}.tar.gz'.format(url, __version__),
keywords=[
'pytorch', 'geometric-deep-learning', 'graph', 'mesh',
'neural-networks', 'spline-cnn'
],
install_requires=install_requires,
setup_requires=setup_requires,
tests_require=tests_require,
packages=find_packages())
You can’t perform that action at this time.