From ed88855f5fa03142d989e90c5ae1588b89858d00 Mon Sep 17 00:00:00 2001
From: Fabian Isensee <f.isensee@dkfz.de>
Date: Wed, 19 Jun 2024 14:00:30 +0200
Subject: [PATCH] use np.allclose as suggested in #2282

---
 nnunetv2/imageio/base_reader_writer.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/nnunetv2/imageio/base_reader_writer.py b/nnunetv2/imageio/base_reader_writer.py
index 2847478ae..4ca536e5d 100644
--- a/nnunetv2/imageio/base_reader_writer.py
+++ b/nnunetv2/imageio/base_reader_writer.py
@@ -21,11 +21,11 @@
 class BaseReaderWriter(ABC):
     @staticmethod
     def _check_all_same(input_list):
-        # compare all entries to the first
-        for i in input_list[1:]:
-            if i != input_list[0]:
-                return False
-        return True
+        if len(input_list) == 1:
+            return True
+        else:
+            # compare all entries to the first
+            return np.allclose(input_list[0], input_list[1:])
 
     @staticmethod
     def _check_all_same_array(input_list):