Skip to content

Commit 364322d

Browse files
committed
fix(sv_split): update multicut test
1 parent b667634 commit 364322d

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

pychunkedgraph/app/segmentation/common.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -472,8 +472,8 @@ def handle_split(table_id):
472472
)
473473
overlap_mask = np.isin(sources_remapped, sinks_remapped)
474474
for sv_to_split in np.unique(sources_remapped[overlap_mask]):
475-
_mask0 = sources_remapped[sources_remapped == sv_to_split]
476-
_mask1 = sinks_remapped[sinks_remapped == sv_to_split]
475+
_mask0 = sources_remapped == sv_to_split
476+
_mask1 = sinks_remapped == sv_to_split
477477
split_supervoxel(
478478
cg,
479479
sv_to_split,

pychunkedgraph/tests/test_uncategorized.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1872,7 +1872,7 @@ def test_path_augmented_multicut(self, sv_data):
18721872
cut_edges_aug = run_multicut(edges, sv_sources, sv_sinks, path_augment=True)
18731873
assert cut_edges_aug.shape[0] == 350
18741874

1875-
with pytest.raises(exceptions.PreconditionError):
1875+
with pytest.raises(exceptions.SupervoxelSplitRequiredError):
18761876
run_multicut(edges, sv_sources, sv_sinks, path_augment=False)
18771877

18781878

0 commit comments

Comments
 (0)