diff --git a/devtools/conda-envs/test_env.yaml b/devtools/conda-envs/test_env.yaml
index b788dea4..08182827 100644
--- a/devtools/conda-envs/test_env.yaml
+++ b/devtools/conda-envs/test_env.yaml
@@ -19,7 +19,8 @@ dependencies:
   - numpy
   - pandas
   - pytorch
-  - lightning<=2.0.1
+  - pydantic<2
+  - lightning
   - matplotlib
   - scikit-learn
   - scipy<1.11
diff --git a/docs/requirements.yaml b/docs/requirements.yaml
index 302d0b56..7e81e9e6 100644
--- a/docs/requirements.yaml
+++ b/docs/requirements.yaml
@@ -15,14 +15,15 @@ dependencies:
   - numpy
   - pandas
   - pytorch
-  - lightning<=2.0.1
+  - pydantic<2 # workaround to avoid clashes with lightning
+  - lightning
 
   # utils
   - matplotlib
   - nbsphinx
   - ipython
   - ipykernel
-  - scipy<1.11
+  - scipy<1.11 # latest version does not work with kdepy
 
   # Pip-only installs
   - pip: