lightning>=2.0.0

[test]
coverage>=5.0
mypy==1.2.0
pytest-cov
pytest>=6.0
scikit-learn<1.2.3,>0.22.1
torchmetrics<0.12.0,>=0.8.0
