diff --git a/pkgs/development/python-modules/finetuning-scheduler/default.nix b/pkgs/development/python-modules/finetuning-scheduler/default.nix index 808c8e1ab6962..35a5a3cbe02ca 100644 --- a/pkgs/development/python-modules/finetuning-scheduler/default.nix +++ b/pkgs/development/python-modules/finetuning-scheduler/default.nix @@ -12,20 +12,22 @@ buildPythonPackage rec { pname = "finetuning-scheduler"; - version = "2.4.0"; + version = "2.5.0"; pyproject = true; - disabled = pythonOlder "3.9"; - src = fetchFromGitHub { owner = "speediedan"; repo = "finetuning-scheduler"; - rev = "refs/tags/v${version}"; - hash = "sha256-uSFGZseSJv519LpaddO6yP6AsIMZutEA0Y7Yr+mEWTQ="; + tag = "v${version}"; + hash = "sha256-neeSATQwAaYN1QGBUXphqqJp9lP3HG2OH4aLdt1cOho="; }; build-system = [ setuptools ]; + pythonRelaxDeps = [ + "pytorch-lightning" + ]; + dependencies = [ pytorch-lightning torch @@ -54,10 +56,12 @@ buildPythonPackage rec { meta = { description = "PyTorch Lightning extension for foundation model experimentation with flexible fine-tuning schedules"; homepage = "https://finetuning-scheduler.readthedocs.io"; - changelog = "https://github.com/speediedan/finetuning-scheduler/blob/${src.rev}/CHANGELOG.md"; + changelog = "https://github.com/speediedan/finetuning-scheduler/blob/${src.tag}/CHANGELOG.md"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ bcdarwin ]; - # "No module named 'torch._C._distributed_c10d'; 'torch._C' is not a package" at import time: - broken = stdenv.hostPlatform.isDarwin; + badPlatforms = [ + # "No module named 'torch._C._distributed_c10d'; 'torch._C' is not a package" at import time: + lib.systems.inspect.patterns.isDarwin + ]; }; } diff --git a/pkgs/development/python-modules/pytorch-lightning/default.nix b/pkgs/development/python-modules/pytorch-lightning/default.nix index b23ea7188d10c..0ba5f707f3559 100644 --- a/pkgs/development/python-modules/pytorch-lightning/default.nix +++ b/pkgs/development/python-modules/pytorch-lightning/default.nix @@ -25,14 +25,14 @@ buildPythonPackage rec { pname = "pytorch-lightning"; - version = "2.4.0"; + version = "2.5.0.post0"; pyproject = true; src = fetchFromGitHub { owner = "Lightning-AI"; repo = "pytorch-lightning"; - rev = "refs/tags/${version}"; - hash = "sha256-FngD3nYTGVEjS1VrXAVps2/B8VlS7BZMDAsmIDtAyW0="; + tag = version; + hash = "sha256-TkwDncyfv1VoV/IErUgF4p0Or5PJbwKoABqo1xXGLVg="; }; preConfigure = '' @@ -68,7 +68,7 @@ buildPythonPackage rec { meta = { description = "Lightweight PyTorch wrapper for machine learning researchers"; homepage = "https://github.com/Lightning-AI/pytorch-lightning"; - changelog = "https://github.com/Lightning-AI/pytorch-lightning/releases/tag/${lib.removePrefix "refs/tags/" src.rev}"; + changelog = "https://github.com/Lightning-AI/pytorch-lightning/releases/tag/${src.tag}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ tbenst ]; };