numpy<=1.19.2,>=1.15
onnxconverter-common<=1.7.0,>=1.6.0
scikit-learn<=0.23.2,>=0.21.3
torch<=1.7.0,>=1.4.*
psutil

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

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

[onnx]
onnxruntime>=1.0.0
onnxmltools>=1.6.0

[sparkml]
pyspark>=2.4.4

[tests]
flake8
pytest
coverage
pre-commit
