From b11854d2f2b807bf74df14c1a2e999699df2f6fd Mon Sep 17 00:00:00 2001
From: Spiros Maggioros <spirosastro@gmail.com>
Date: Thu, 12 Dec 2024 02:18:12 +0200
Subject: [PATCH] require nnunetv2==2.5.1

---
 requirements.txt | 2 +-
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 16bc09e..b53f4e7 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 torch==2.2.1
-nnunetv2==2.2.1
+nnunetv2==2.5.1
 argparse
 huggingface_hub
 pathlib
diff --git a/setup.py b/setup.py
index 331fedb..6510cf8 100644
--- a/setup.py
+++ b/setup.py
@@ -23,7 +23,7 @@
     python_requires=">=3.8",
     install_requires=[
         "torch<=2.2.1",
-        "nnunetv2<=2.2.1",
+        "nnunetv2<=2.5.1",
         "argparse",
         "huggingface_hub"
     ],