From 9b46c7ea31ee1dca7fc8852b0df2880150fed061 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 5 Nov 2025 21:31:18 +0000
Subject: [PATCH 1/2] python3Packages.pytorch-lightning: 2.5.5 -> 2.5.6

Diff: https://github.com/Lightning-AI/pytorch-lightning/compare/2.5.5...2.5.6

Changelog: https://github.com/Lightning-AI/pytorch-lightning/releases/tag/2.5.6
---
 pkgs/development/python-modules/pytorch-lightning/default.nix | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pkgs/development/python-modules/pytorch-lightning/default.nix b/pkgs/development/python-modules/pytorch-lightning/default.nix
index d8e90c7c0cc2..979f6c4b0fa8 100644
--- a/pkgs/development/python-modules/pytorch-lightning/default.nix
+++ b/pkgs/development/python-modules/pytorch-lightning/default.nix
@@ -24,14 +24,14 @@
 
 buildPythonPackage rec {
   pname = "pytorch-lightning";
-  version = "2.5.5";
+  version = "2.5.6";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "Lightning-AI";
     repo = "pytorch-lightning";
     tag = version;
-    hash = "sha256-8CDVvgaxnFWO4Fl5lW/+cn/1WZCgVXYys86iOVNYUfY=";
+    hash = "sha256-ojmE0d6Wy4UqQu4kBBE2qtQ4AYqplHOB7wJ7hEte664=";
   };
 
   preConfigure = ''

From 2c42578129d67a6a2319986ea1c8bf0f7b39bdfb Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sat, 8 Nov 2025 22:53:11 +0000
Subject: [PATCH 2/2] python3Packages.finetuning-scheduler: fix
 ReduceLROnPlateau import

https://github.com/speediedan/finetuning-scheduler/pull/21
---
 .../python-modules/finetuning-scheduler/default.nix       | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/pkgs/development/python-modules/finetuning-scheduler/default.nix b/pkgs/development/python-modules/finetuning-scheduler/default.nix
index 113e77b2dca2..85b46d882a52 100644
--- a/pkgs/development/python-modules/finetuning-scheduler/default.nix
+++ b/pkgs/development/python-modules/finetuning-scheduler/default.nix
@@ -27,6 +27,14 @@ buildPythonPackage rec {
     hash = "sha256-AfkrWuqpFS71Zrh5NsamzxMitKCsqPF50F9zTDdDhRg=";
   };
 
+  # See https://github.com/speediedan/finetuning-scheduler/pull/21
+  postPatch = ''
+    substituteInPlace src/finetuning_scheduler/strategy_adapters/base.py \
+      --replace-fail \
+        "from lightning.fabric.utilities.types import ReduceLROnPlateau" \
+        "from torch.optim.lr_scheduler import ReduceLROnPlateau"
+  '';
+
   build-system = [ setuptools ];
 
   pythonRelaxDeps = [