dill
numpy>=1.15
onnxconverter-common>=1.6.0
protobuf>=3.20.2
psutil
scikit-learn
scipy
torch>1.7.0

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

[extra]
lightgbm>=2.2
prophet==1.1
xgboost>=0.90

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

[sparkml]
pyarrow>1.0
pyspark<3.1.2,>=2.4.4

[tests]
coverage
flake8
pre-commit
pytest
