File size: 705 Bytes
cf84204 | 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 | """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",
],
},
)
|