lightning>=2.0.0

[test]
coverage>=5.0
mypy==1.1.1
pytest-cov
pytest>=6.0
scikit-learn<1.2.1,>0.22.1
torchmetrics<0.11.0,>=0.8.0
