Skip to content

Commit 676c791

Browse files
committed
Fix segmentation node removal.
1 parent 47bf4be commit 676c791

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

genesis/vis/rasterizer_context.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -875,7 +875,7 @@ def create_node_seg(self, seg_key, seg_node):
875875
self.seg_node_map[seg_node] = self.seg_idxc_to_idxc_rgb(seg_idxc)
876876

877877
def remove_node_seg(self, seg_node):
878-
self.seg_node_map.pop(seg_node)
878+
self.seg_node_map.pop(seg_node, None)
879879

880880
def generate_seg_vars(self):
881881
# seg_key: same as entity/link/geom's idx

0 commit comments

Comments
 (0)