diff --git a/pkgs/development/python-modules/torchmetrics/default.nix b/pkgs/development/python-modules/torchmetrics/default.nix index e5c05c6dd497..5cc57ed0bed3 100644 --- a/pkgs/development/python-modules/torchmetrics/default.nix +++ b/pkgs/development/python-modules/torchmetrics/default.nix @@ -8,6 +8,7 @@ , scikit-learn , scikit-image , packaging +, pretty-errors , psutil , py-deprecate , torch @@ -20,7 +21,7 @@ let pname = "torchmetrics"; - version = "1.3.2"; + version = "1.4.0"; in buildPythonPackage { inherit pname version; @@ -32,13 +33,14 @@ buildPythonPackage { owner = "Lightning-AI"; repo = "torchmetrics"; rev = "refs/tags/v${version}"; - hash = "sha256-0lverGF8HVlnwWmnCjScWoSfWPLZlMF2cHDG+uxNCCQ="; + hash = "sha256-5ANEpUdd5eyzBhhAEHrvHNHMOQuXhkkKuL1t3dJLhdE="; }; - propagatedBuildInputs = [ + dependencies = [ numpy lightning-utilities packaging + pretty-errors py-deprecate ];