From cdcfe3d39a16f56177cc42cc94f8a931db94cc76 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Thu, 31 Oct 2024 06:57:53 +0100
Subject: [PATCH] Don't allow appending different data types
---
model_server/base/accessors.py | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/model_server/base/accessors.py b/model_server/base/accessors.py
index 0eb22a65..4e3db82d 100644
--- a/model_server/base/accessors.py
+++ b/model_server/base/accessors.py
@@ -181,6 +181,8 @@ class GenericImageDataAccessor(ABC):
}
def append_channels(self, acc):
+ if self.dtype != acc.dtype:
+ raise DataTypeError(f'Cannot append data of type {acc.dtype} to an accessor with type {self.dtype}')
return self._derived_accessor(
np.concatenate(
(self.data, acc.data),
@@ -570,6 +572,9 @@ class FileNotFoundError(Error):
class DataShapeError(Error):
pass
+class DataTypeError(Error):
+ pass
+
class FileWriteError(Error):
pass
--
GitLab