diff --git a/pyproject.toml b/pyproject.toml
index b6f1fe122050f49b99a0d05e87708711fa7c630f..5d21fb6ce21ef79ddfc736cd72f8f46e649ba679 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -41,8 +41,7 @@ dependencies = [
   "opencv-python",
   "torch>=1.8",
   "torchvision>=0.10",
-  "lightning <2.1.0a0,>=2.0.3",
-  "pydantic <2.0,>=1.7.4", # temporary, until issue #31 is fixed
+  "lightning <2.2.0a0,>=2.1.0",
   "tensorboard",
   "grad-cam>=1.4.8",
 ]