torch>=1.9

[build]
setuptools
wheel
twine

[dev]
black
flake8
flake8-black
isort>=5.7
pytest
pytest-cov
ptflops>=0.6
numpy
pytorch_lightning

[docs]
sphinx>=3.0
nbsphinx>=0.8
pandoc>=1.0
docutils>=0.16
sphinxcontrib-fulltoc>=1.0
sphinxcontrib-mockautodoc
sphinx-autodoc-typehints>=1.0
sphinx-paramlinks>=0.4.0
sphinx-togglebutton>=0.2
sphinx-copybutton>=0.3
sphinx-autoapi>=1.7
m2r2>=0.2
