From d6f46e60b01ed28b3f333566b05b21770a61b593 Mon Sep 17 00:00:00 2001
From: Ben Darwin <bcdarwin@gmail.com>
Date: Tue, 31 Mar 2026 23:59:17 -0400
Subject: [PATCH] python313Packages.torchio: 0.21.2 -> 1.0.1

---
 .../python-modules/torchio/default.nix        | 34 ++++++++++++-------
 1 file changed, 22 insertions(+), 12 deletions(-)

diff --git a/pkgs/development/python-modules/torchio/default.nix b/pkgs/development/python-modules/torchio/default.nix
index 7060b38a1a88..fe56d66701e8 100644
--- a/pkgs/development/python-modules/torchio/default.nix
+++ b/pkgs/development/python-modules/torchio/default.nix
@@ -11,6 +11,7 @@
   deprecated,
   einops,
   humanize,
+  jaxtyping,
   nibabel,
   numpy,
   packaging,
@@ -24,6 +25,7 @@
   # optional dependencies
   colorcet,
   matplotlib,
+  monai,
   pandas,
   ffmpeg-python,
   scikit-learn,
@@ -35,14 +37,14 @@
 
 buildPythonPackage rec {
   pname = "torchio";
-  version = "0.21.2";
+  version = "1.0.1";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "TorchIO-project";
     repo = "torchio";
     tag = "v${version}";
-    hash = "sha256-HyhoF3F4EsJCOkWmoQkFqNsx8YbBsUFlQC+AggaXLXI=";
+    hash = "sha256-b6ED3IyQgC0A0dqYiXC0GTf6ZqwE1Ka7ojM7OJu1xfI=";
   };
 
   build-system = [
@@ -53,6 +55,7 @@ buildPythonPackage rec {
     deprecated
     einops
     humanize
+    jaxtyping
     nibabel
     numpy
     packaging
@@ -64,21 +67,28 @@ buildPythonPackage rec {
     typer
   ];
 
-  optional-dependencies = {
-    csv = [ pandas ];
-    plot = [
-      colorcet
-      matplotlib
-    ];
-    video = [ ffmpeg-python ];
-    sklearn = [ scikit-learn ];
-  };
+  optional-dependencies =
+    let
+      extras = {
+        csv = [ pandas ];
+        monai = [ monai ];
+        plot = [
+          colorcet
+          matplotlib
+        ];
+        video = [ ffmpeg-python ];
+        sklearn = [ scikit-learn ];
+      };
+    in
+    extras // { all = lib.concatLists (lib.attrValues extras); };
 
   nativeCheckInputs = [
     matplotlib
     parameterized
     pytestCheckHook
-  ];
+  ]
+  ++ optional-dependencies.monai
+  ++ optional-dependencies.sklearn;
 
   disabledTests = [
     # tries to download models: