numpy<3,>=1.22

[dev]
pytest>=8
scikit-learn>=1.3

[huggingface]
transformers>=4.40
datasets>=2.19

[sklearn]
scikit-learn>=1.3

[tensorflow]
tensorflow>=2.15

[torch]

[torch:sys_platform != "darwin" or platform_machine == "arm64"]
torch>=2.3

[torch:sys_platform == "darwin" and platform_machine == "x86_64"]
torch<2.3,>=2.0
numpy<2,>=1.22
