numpy<2
torch==2.3.0
torchdata==0.7.1
dgl==2.2.1
gensim
scikit_learn
setuptools
pydantic
pyyaml


