dataclasses-json
numpy
pykeen
pykeen>=1.10.1
scipy>=1.7.0
click
click_default_group
scikit-learn
torch<2.4,>=2.0
torchvision<0.17.0,>=0.15.0
tqdm
requests
optuna>=2.0.0
pandas>=1.0.0
tabulate
more_click
more_itertools
pystow>=0.4.3
docdata
class_resolver>0.4.2
pyyaml
torch_max_mem>=0.1.1
torch-ppr>=0.0.7
typing_extensions

[:sys_platform == "win32"]
numpy<2.0.0
