torch>=1.8.1
packaging

[all]
accelerate==0.5.1
pytest==6.2.5
pytest-flake8==1.0.7
flake8==3.9.2
pytest-cov==3.0.0
pytest-mypy==0.8.1
pytest-xdist==2.4.0
autopep8
pydocstyle
pre-commit>=2.0
yapf
isort
numpy
scipy

[all:python_version < "3.7"]
dataclasses

[dev]
pytest==6.2.5
pytest-flake8==1.0.7
flake8==3.9.2
pytest-cov==3.0.0
pytest-mypy==0.8.1
pytest-xdist==2.4.0
autopep8
pydocstyle
pre-commit>=2.0
yapf
isort
numpy
scipy

[x]
accelerate==0.5.1

[x:python_version < "3.7"]
dataclasses
