matplotlib==3.9.2
numpy==1.23.1
pandas==2.2.3
rich_argparse_plus==0.3.1.4
scikit_learn==1.5.2
torch==1.12.1
torch_cluster==1.6.0
