numpy>=1.18.0
pandas>=0.25.3
ujson
torch==1.6.0
toml>=0.10.2
tqdm
typing-extensions

[:python_version == "3.7"]
scikit-learn<1.1.0,>=0.24.0
scipy<1.8.0,>=1.4.1

[:python_version == "3.8"]
scikit-learn>=0.24.0
scipy>=1.4.1
