try:
    import torch
except ModuleNotFoundError:
    raise ModuleNotFoundError(
        "PyTorch not installed. "
        "Please appropriately install PyTorch, "
        "see https://pytorch.org/ for installation."
    )
try:
    import torch_scatter
    import torch_sparse
    import torch_cluster
    import torch_spline_conv
    import torch_geometric
except ModuleNotFoundError:
    raise ModuleNotFoundError(
        "PyTorch-Geometric not fully installed. "
        "Please appropriately install PyTorch-Geometric, "
        "see https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html for installation."
    )

from setuptools import setup, find_packages

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

''' https://packaging.python.org/guides/distributing-packages-using-setuptools/ '''
''' https://setuptools.readthedocs.io/en/latest/ '''
setup(
    name='auto-graph-learning',
    version='0.1.1',
    author='THUMNLab/aglteam',
    maintainer='THUMNLab/aglteam',
    author_email='xin_wang@tsinghua.edu.cn',
    description='AutoML tools for graph-structure dataset',
    long_description=long_description,
    long_description_content_type='text/markdown',
    include_package_data=True,
    packages=find_packages(),
    # https://packaging.python.org/guides/distributing-packages-using-setuptools/#python-requires
    python_requires='~=3.6',
    # https://pypi.org/classifiers/
    classifiers=[
        "Development Status :: 2 - Pre-Alpha",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9"
    ],
    # https://setuptools.readthedocs.io/en/latest/userguide/dependency_management.html
    # note that setup_requires and tests_require are deprecated
    install_requires=[
        'bayesian-optimization',
        'chocolate',
        'dill',
        'hyperopt',
        'lightgbm',
        'networkx',
        'numpy',
        'netlsd',
        'ogb',
        'psutil',
        'pyyaml',
        'requests',
        'scikit-learn',
        'scipy',
        'tabulate',
        'torch',
        'torch-geometric',
        'torch-scatter',
        'torch-sparse',
        'torch-cluster',
        'torch-spline-conv',
        'tqdm'
    ]
)