from setuptools import find_packages, setup

setup(
    name='torch_geometric_benchmark',
    version='0.1.0',
    description='PyG Benchmark Suite',
    author='Matthias Fey',
    author_email='matthias.fey@tu-dortmund.de',
    url='https://github.com/pyg-team/pytorch_geometric_benchmark',
    install_requires=['scikit-learn'],
    packages=find_packages(),
)