From be6040f368704700ba69e80b55479d5b6e8e6b43 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Sat, 2 Nov 2024 08:11:00 +0100
Subject: [PATCH] Int64 type was causing issues downstream in eval/query calls;
 Int16 is adequate

---
 model_server/base/roiset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py
index e8184712..925b3b7b 100644
--- a/model_server/base/roiset.py
+++ b/model_server/base/roiset.py
@@ -1152,7 +1152,7 @@ class RoiSetWithDerivedChannels(RoiSet):
             self.get_patch_masks_acc(expanded=False, pad_to=None)
         )
 
-        self._df['classify_by_' + name] = pd.Series(dtype='Int64')
+        self._df['classify_by_' + name] = pd.Series(dtype='Int16')
 
         for i, roi in enumerate(self):
             oc = np.unique(
-- 
GitLab