fix: no mesh edge case

CAVEconnectome · Apr 19, 2023 · 7a93c77 · 7a93c77
1 parent b10ecb4
commit 7a93c77
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/pychunkedgraph/meshing/manifest/multiscale.py b/pychunkedgraph/meshing/manifest/multiscale.py
@@ -104,15 +104,16 @@ def build_octree(
         octree[offset + 4] = end_empty
 
         octree_node_ids[row_counter] = current_node
-        if children.size == 1:
-            # map to child fragment
-            octree_fragments[row_counter] = mesh_fragments[children[0]]
-        else:
-            try:
+
+        try:
+            if children.size == 1:
+                # map to child fragment
+                octree_fragments[row_counter] = mesh_fragments[children[0]]
+            else:
                 octree_fragments[row_counter] = mesh_fragments[current_node]
-            except KeyError:
-                # mark node empty
-                octree[offset + 4] |= 1 << 31
+        except KeyError:
+            # no mesh, mark node empty
+            octree[offset + 4] |= 1 << 31
 
         for child in children:
             que.append((child, current_depth + 1))