From 058633cb89729a62183ea7d8fa2f5d935a22059f Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 25 Mar 2024 10:41:39 +0100
Subject: [PATCH] Set default values for time, color, and z-axes in ilastik
model validation
---
model_server/extensions/ilastik/models.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/model_server/extensions/ilastik/models.py b/model_server/extensions/ilastik/models.py
index d17e6954..ac322e85 100644
--- a/model_server/extensions/ilastik/models.py
+++ b/model_server/extensions/ilastik/models.py
@@ -80,7 +80,11 @@ class IlastikPixelClassifierModel(IlastikModel, SemanticSegmentationModel):
ax = raw_info['axistags'][()]
ax_keys = [ax['key'].upper() for ax in json.loads(ax)['axes']]
shape = raw_info['shape'][()]
- return dict(zip(ax_keys, shape))
+ dd = dict(zip(ax_keys, shape))
+ for ci in 'TCZ':
+ if ci not in dd.keys():
+ dd[ci] = 1
+ return dd
@property
def model_chroma(self):
--
GitLab