numpy<=1.19.4,>=1.15
onnxconverter-common<=1.7.0,>=1.6.0
scipy<=1.5.4
scikit-learn<=0.23.2,>=0.21.3
torch<=1.8.0,>=1.4.*
psutil
dill

[benchmark]
onnxruntime<1.7.0,>=1.0.0
onnxmltools>=1.6.0
skl2onnx<=1.7.0
xgboost>=0.90
lightgbm<3,>=2.2
memory-profiler
psutil

[extra]
xgboost>=0.90
lightgbm<3,>=2.2

[onnx]
onnxruntime<1.7.0,>=1.0.0
onnxmltools>=1.6.0
skl2onnx<=1.7.0

[sparkml]
pyspark>=2.4.4

[tests]
flake8
pytest
coverage
pre-commit
