diff --git a/pkgs/development/python-modules/torchio/default.nix b/pkgs/development/python-modules/torchio/default.nix index a5c3ad52f795..941bd288dfe0 100644 --- a/pkgs/development/python-modules/torchio/default.nix +++ b/pkgs/development/python-modules/torchio/default.nix @@ -10,7 +10,7 @@ # dependencies deprecated, einops, - matplotlib, + humanize, nibabel, numpy, packaging, @@ -21,22 +21,28 @@ tqdm, typer, + # optional dependencies + colorcet, + matplotlib, + pandas, + ffmpeg-python, + scikit-learn, + # tests - humanize, parameterized, pytestCheckHook, }: buildPythonPackage rec { pname = "torchio"; - version = "0.20.21"; + version = "0.20.22"; pyproject = true; src = fetchFromGitHub { owner = "TorchIO-project"; repo = "torchio"; tag = "v${version}"; - hash = "sha256-l2KQLZDxsP8Bjk/vPG2YbU+8Z6/lOvNvy9NYKTdW+cY="; + hash = "sha256-LP0hlle8BCoZrJWs5aX/xvI+EPHdOGBARoKwQRqswQc="; }; build-system = [ @@ -58,6 +64,16 @@ buildPythonPackage rec { typer ]; + optional-dependencies = { + csv = [ pandas ]; + plot = [ + colorcet + matplotlib + ]; + video = [ ffmpeg-python ]; + sklearn = [ scikit-learn ]; + }; + nativeCheckInputs = [ matplotlib parameterized