Skip to content
Snippets Groups Projects
Commit 6635097d authored by Alán Muñoz's avatar Alán Muñoz
Browse files

fix(extractor): adjust number of cell labels

parent 7861f7d6
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
tests/aliby/network/extraction/test_base.py
+ 1
− 1
View file @ 6635097d
...@@ -52,7 +52,7 @@ def test_extractor(imgs, masks, tree): ...@@ -52,7 +52,7 @@ def test_extractor(imgs, masks, tree):
extractor.tree = tree extractor.tree = tree
traps = imgs["GFP"] traps = imgs["GFP"]
# Generate mock labels # Generate mock labels
labels = list(range(masks.shape[2])) labels = list(range(len(masks)))
for ch_branches in extractor.params.tree.values(): for ch_branches in extractor.params.tree.values():
print( print(
extractor.reduce_extract( extractor.reduce_extract(
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment