"""Setup script for GraPHFormer."""

from setuptools import setup, find_packages

setup(
    name="graphformer",
    version="1.0.0",
    description="GraPHFormer: Graph-Persistence Hybrid Transformer for Neuron Morphology",
    author="",
    packages=find_packages(),
    python_requires=">=3.8",
    install_requires=[
        "torch>=1.10.0",
        "torchvision>=0.11.0",
        "dgl>=0.8.0",
        "numpy>=1.20.0",
        "scikit-learn>=1.0.0",
        "nltk>=3.6.0",
        "tqdm>=4.60.0",
        "networkx>=2.6.0",
        "Pillow>=8.0.0",
    ],
    extras_require={
        "dev": [
            "pytest>=6.0.0",
            "black>=21.0",
            "flake8>=3.9.0",
        ],
    },
)