BUG: Fix arrayFromSegmentBinaryLabelmap leaving temporary node behind

slicer.util.arrayFromSegmentBinaryLabelmap left temporary color node in the scene. This commit removes this unneeded node.
Slicer · Dec 8, 2022 · 5a3f338 · 5a3f338
1 parent 2ed9451
commit 5a3f338
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Base/Python/slicer/util.py b/Base/Python/slicer/util.py
@@ -2045,6 +2045,8 @@ def arrayFromSegmentBinaryLabelmap(segmentationNode, segmentId, referenceVolumeN
             raise RuntimeError("Export of segment failed.")
         narray = slicer.util.arrayFromVolume(labelmapVolumeNode)
     finally:
+        if labelmapVolumeNode.GetDisplayNode():
+            slicer.mrmlScene.RemoveNode(labelmapVolumeNode.GetDisplayNode().GetColorNode())
         slicer.mrmlScene.RemoveNode(labelmapVolumeNode)
 
     return narray