diff --git a/setup.cfg b/setup.cfg index ff175d3..86e33c3 100644 --- a/setup.cfg +++ b/setup.cfg @@ -22,7 +22,7 @@ install_requires = importlib-resources>=5.10.0 joblib>=1.1.0 numpy>=1.16.5 - pytorch_lightning<=1.9.5 + pytorch_lightning>=2.0.0 pyyaml>=5.4.1 regex>=2.5.91 scikit-learn>=1.0.0