diff --git a/pkgs/development/python-modules/finetuning-scheduler/default.nix b/pkgs/development/python-modules/finetuning-scheduler/default.nix index d4a2752c8ea3..0ff7318f0bb3 100644 --- a/pkgs/development/python-modules/finetuning-scheduler/default.nix +++ b/pkgs/development/python-modules/finetuning-scheduler/default.nix @@ -12,16 +12,16 @@ buildPythonPackage rec { pname = "finetuning-scheduler"; - version = "2.3.3"; + version = "2.4.0"; pyproject = true; - disabled = pythonOlder "3.8"; + disabled = pythonOlder "3.9"; src = fetchFromGitHub { owner = "speediedan"; repo = "finetuning-scheduler"; rev = "refs/tags/v${version}"; - hash = "sha256-Nci+NcfcaZV3NH3MTCyr8PwPdDykJTdq8+CmtngWdiM="; + hash = "sha256-uSFGZseSJv519LpaddO6yP6AsIMZutEA0Y7Yr+mEWTQ="; }; build-system = [ setuptools ]; diff --git a/pkgs/development/python-modules/pytorch-lightning/default.nix b/pkgs/development/python-modules/pytorch-lightning/default.nix index bcea697b8325..116997ceb8e4 100644 --- a/pkgs/development/python-modules/pytorch-lightning/default.nix +++ b/pkgs/development/python-modules/pytorch-lightning/default.nix @@ -2,12 +2,16 @@ lib, buildPythonPackage, fetchFromGitHub, + + # build-system + setuptools, + + # dependencies fsspec, lightning-utilities, numpy, packaging, pyyaml, - setuptools, tensorboardx, torch, torchmetrics, @@ -21,14 +25,14 @@ buildPythonPackage rec { pname = "pytorch-lightning"; - version = "2.3.3"; + version = "2.4.0"; pyproject = true; src = fetchFromGitHub { owner = "Lightning-AI"; repo = "pytorch-lightning"; rev = "refs/tags/${version}"; - hash = "sha256-JgT+OiBToBBa3h556gF+kmcqLKLaK17tlTPokvEDyUE="; + hash = "sha256-FngD3nYTGVEjS1VrXAVps2/B8VlS7BZMDAsmIDtAyW0="; }; preConfigure = '' @@ -39,12 +43,12 @@ buildPythonPackage rec { dependencies = [ fsspec + lightning-utilities numpy packaging pyyaml tensorboardx torch - lightning-utilities torchmetrics tqdm traitlets @@ -61,11 +65,11 @@ buildPythonPackage rec { pythonImportsCheck = [ "pytorch_lightning" ]; - meta = with lib; { + 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/${src.rev}"; - license = licenses.asl20; - maintainers = with maintainers; [ tbenst ]; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ tbenst ]; }; }