Merge pull request #225771 from bcdarwin/python3-torchio

python310Packages.torchio: init at 0.18.90

pkgs/development/python-modules/torchio/default.nix

@@ -0,0 +1,64 @@
+{ stdenv
+, lib
+, buildPythonPackage
+, fetchFromGitHub
+, pytestCheckHook
+, pythonOlder
+, deprecated
+, humanize
+, matplotlib
+, nibabel
+, numpy
+, parameterized
+, scipy
+, simpleitk
+, torch
+, tqdm
+, typer
+}:
+
+buildPythonPackage rec {
+  pname = "torchio";
+  version = "0.18.90";
+  format = "pyproject";
+  disabled = pythonOlder "3.7";
+
+  src = fetchFromGitHub {
+    owner = "fepegar";
+    repo = pname;
+    rev = "refs/tags/v${version}";
+    hash = "sha256-h8cvNhOkjMMbQ6Nry8FKtwnK+yhRYRGjXi/xp0i5yyY=";
+  };
+
+  propagatedBuildInputs = [
+    deprecated
+    humanize
+    nibabel
+    numpy
+    scipy
+    simpleitk
+    torch
+    tqdm
+    typer
+  ] ++ typer.passthru.optional-dependencies.all;
+
+  nativeCheckInputs = [ pytestCheckHook matplotlib parameterized ];
+  disabledTests = [
+    # tries to download models:
+    "test_load_all"
+  ] ++ lib.optionals stdenv.isAarch64 [
+    # RuntimeError: DataLoader worker (pid(s) <...>) exited unexpectedly
+    "test_queue_multiprocessing"
+  ];
+  pythonImportsCheck = [
+    "torchio"
+    "torchio.data"
+  ];
+
+  meta = with lib; {
+    description = "Medical imaging toolkit for deep learning";
+    homepage = "http://www.torchio.org/";
+    license = licenses.asl20;
+    maintainers = [ maintainers.bcdarwin ];
+  };
+}

pkgs/top-level/python-packages.nix

@@ -11945,6 +11945,8 @@ self: super: with self; {
 
   torchmetrics = callPackage ../development/python-modules/torchmetrics { };
 
+  torchio = callPackage ../development/python-modules/torchio { };
+
   torchinfo = callPackage ../development/python-modules/torchinfo { };
 
   torchlibrosa = callPackage ../development/python-modules/torchlibrosa { };