einops<1.0.0,>=0.6.0
torch<3.0.0,>=1.7.0

[all]
black
flake8
isort
mypy
pytest
pytest-cov
pre-commit

[dev]
pre-commit
black
flake8
isort
mypy
pytest
pytest-cov

[test]
black
flake8
isort
mypy
pytest
pytest-cov
