From 7b37e73d872c77276fc94eb194ddfd5197b8d008 Mon Sep 17 00:00:00 2001
From: Ben Darwin <bcdarwin@gmail.com>
Date: Fri, 31 Mar 2023 17:12:34 -0400
Subject: [PATCH] python310Packages.torchio: init at 0.18.90

---
 .../python-modules/torchio/default.nix        | 64 +++++++++++++++++++
 pkgs/top-level/python-packages.nix            |  2 +
 2 files changed, 66 insertions(+)
 create mode 100644 pkgs/development/python-modules/torchio/default.nix

diff --git a/pkgs/development/python-modules/torchio/default.nix b/pkgs/development/python-modules/torchio/default.nix
new file mode 100644
index 000000000000..58ab010eb969
--- /dev/null
+++ b/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 ];
+  };
+}
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 2594f7e87272..c67a1eba8e4c 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -11901,6 +11901,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 { };