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