# pytest 6.0 has linter-breaking changes
pytest>=6.0.1,<8.0.0 # Versions 8.0.0 and above require Python 3.9 or above.
mypy==0.910
# responses 0.23.2 requires urllib3>=2, which breaks several other dependencies
responses!=0.23.2
requests_mock
coverage
deepspeed==0.8.3
# lightning not tested but required for linter checks
lightning
transformers>=4.8.2,<4.29.0
# Intel MAC only supports torch <= 2.2.2
torch==2.2.2; sys_platform == 'darwin' and platform_machine == 'x86_64'
torch==2.3.0; sys_platform != 'darwin' or platform_machine != 'x86_64'
torchvision==0.17.2; sys_platform == 'darwin' and platform_machine == 'x86_64'
torchvision==0.18.0; sys_platform != 'darwin' or platform_machine != 'x86_64'
tensorflow==2.12.0; sys_platform != 'darwin' or platform_machine != 'arm64'
tensorflow-macos==2.11.0; sys_platform == 'darwin' and platform_machine == 'arm64'
attrdict3
moto<5.0.0 # moto 5.0.0 drops mock_s3, need to replace with mock_aws
# Pydantic V2 has changes that break existing tests
pydantic<2
pandas
