requirements.txt

ase==3.22.1
cached-property==1.5.2
certifi==2023.7.22
chardet==5.1.0
cycler==0.11.0
decorator
future==0.18.3
googledrivedownloader==0.4
h5py==3.9.0
idna==3.4
imblearn
isodate==0.6.1
Jinja2==3.1.2
joblib==1.3.1
kiwisolver==1.4.4
llvmlite==0.40.1
MarkupSafe==2.1.3
matplotlib==3.7.2
networkx
nilearn==0.10.1
nni==2.10.1
node2vec~=0.4.6
numba
numpy
pandas==2.0.3
Pillow==10.0.0
pyg-lib -f https://data.pyg.org/whl/torch-2.0.1+cu118.html
pymanopt==2.1.1
pyparsing==3.0.0
python-dateutil==2.8.2
python-louvain==0.16
pytz==2023.3
rdflib==6.3.2
requests==2.31.0
scikit-learn==1.3.0
scipy==1.11.1
six==1.16.0
threadpoolctl==3.2.0
torch==2.0.1 #--index-url https://download.pytorch.org/whl/cu118

#PyG
-f https://data.pyg.org/whl/torch-2.0.1+cu118.html
torch_geometric==2.3.1
torch-cluster #--index-url https://pytorch-geometric.com/whl/torch-{2.0.1}+{cu118}.html
torch-scatter #--index-url https://pytorch-geometric.com/whl/torch-{2.0.1}+{cu118}.html
torch-sparse #--index-url https://pytorch-geometric.com/whl/torch-{2.0.1}+{cu118}.html
torch-spline-conv #--index-url https://pytorch-geometric.com/whl/torch-{2.0.1}+{cu118}.html

tqdm==4.65.0
typing-extensions==4.7.1
urllib3==2.0.4