From 6635097d34100a23c77a237e026db8d6ee9897ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Al=C3=A1n=20Mu=C3=B1oz?= <alan.munoz@ed.ac.uk>
Date: Mon, 27 Feb 2023 12:04:14 +0000
Subject: [PATCH] fix(extractor): adjust number of cell labels
---
tests/aliby/network/extraction/test_base.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tests/aliby/network/extraction/test_base.py b/tests/aliby/network/extraction/test_base.py
index f1d079d2..59ebc976 100644
--- a/tests/aliby/network/extraction/test_base.py
+++ b/tests/aliby/network/extraction/test_base.py
@@ -52,7 +52,7 @@ def test_extractor(imgs, masks, tree):
extractor.tree = tree
traps = imgs["GFP"]
# Generate mock labels
- labels = list(range(masks.shape[2]))
+ labels = list(range(len(masks)))
for ch_branches in extractor.params.tree.values():
print(
extractor.reduce_extract(
--
GitLab