[FIX] Patch for pytorch-lightning (#1513)

deven367 · web-flow · commit 690317fb3b4a · 2026-05-06T12:51:47.000-04:00
diff --git a/docs/models.stemgnn.html.md b/docs/models.stemgnn.html.md
@@ -24,7 +24,7 @@ Zhang (2020). “Spectral Temporal Graph Neural Network for Multivariate
 Time-series
 Forecasting”.](https://proceedings.neurips.cc/paper/2020/hash/cdf6581cb7aca4b7e19ef136c6e601a5-Abstract.html)
 
-![Figure 1. StemGNN.](imgs_models/stemgnn.png)
+![Figure 1. StemGNN.](imgs_models/StemGNN.png)
 *Figure 1. StemGNN.*
 
 ## 1. StemGNN
diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ dependencies = [
     "scipy",
     "torch>=2.4.0",
     "tornado>=6.5.5",
-    "pytorch-lightning>=2.0.0",
+    "pytorch-lightning>=2.0.0,<2.6.0",
     "ray[train,tune]>=2.2.0",
     "optuna",
     "utilsforecast>=0.2.3",
diff --git a/uv.lock b/uv.lock
