python3Packages.finetuning-scheduler: fix ReduceLROnPlateau import

https://github.com/speediedan/finetuning-scheduler/pull/21
2025-11-08 22:53:14 +00:00
parent 9b46c7ea31
commit 2c42578129
1 changed files with 8 additions and 0 deletions
@@ -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 = [