Skip to content

Commit 52fc3b0

Browse files
committed
fix(tests): match shape for np isin
1 parent 3f6f2e2 commit 52fc3b0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pychunkedgraph/graph/cutting.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -395,8 +395,8 @@ def _remap_cut_edge_set(self, cut_edge_set):
395395

396396
remapped_cutset = np.array(remapped_cutset, dtype=np.uint64)
397397

398-
remapped_cutset_flattened_view = remapped_cutset.view(dtype="u8,u8")
399-
edges_flattened_view = self.cg_edges.view(dtype="u8,u8")
398+
remapped_cutset_flattened_view = remapped_cutset.view(dtype="u8,u8").ravel()
399+
edges_flattened_view = self.cg_edges.view(dtype="u8,u8").ravel()
400400

401401
cutset_mask = np.isin(remapped_cutset_flattened_view, edges_flattened_view)
402402

0 commit comments

Comments
 (0)