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] 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 = [