torch<=1.10
di-engine<0.4.1
scikit-learn>=0.18.0
protobuf<=3.20.1

[doc]
sphinx>=2.2.1
sphinx_rtd_theme~=0.4.3
enum_tools
sphinx-toolbox
recommonmark
sphinx-multiversion~=0.2.4

[style]
yapf==0.29.0
flake8~=3.5

[test]
pytest~=6.2.5
pytest-cov~=3.0.0
pytest-mock~=3.6.1
pytest-xdist>=1.34.0
pytest-rerunfailures~=10.2
pytest-timeout~=2.0.2
pytest-benchmark~=3.4.0
