From 49581d11955a67a73941f9a9a5267f8e819d36d3 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 25 Mar 2024 17:05:31 +0100
Subject: [PATCH] Temporary error-handling for debug...

---
 model_server/base/roiset.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py
index 7cd327ae..33c30c12 100644
--- a/model_server/base/roiset.py
+++ b/model_server/base/roiset.py
@@ -290,11 +290,16 @@ class RoiSet(object):
 
         # assign labels to object map:
         for i, roi in enumerate(self):
-            oc = np.unique(
-                mask_largest_object(
-                    obmap_patches.iat(i).data
-                )
-            )[-1]
+            try:
+                oc = np.unique(
+                    mask_largest_object(
+                        obmap_patches.iat(i).data
+                    )
+                )[-1]
+            except Exception as e:
+                print(i)
+                print(np.unique(mask_largest_object(obmap_patches.iat(i).data)))
+                raise e
             self._df.loc[roi.Index, 'classify_by_' + name] = oc
             om[self.acc_obj_ids.data == roi.label] = oc
         self.object_class_maps[name] = InMemoryDataAccessor(om)
-- 
GitLab