lightning-utilities>=0.7.0

[lightning]
lightning>=2.0.0

[pytorch-lightning]
pytorch-lightning>=2.0.0

[test]
coverage>=5.0
pytest-cov
pytest>=6.0
scikit-learn<1.4.0,>0.22.1
torchmetrics<1.3.0,>=0.8.0

[typing]
mypy==1.5.1
pytorch-lightning==2.0.9
