Set parents correctly on split nodes

This commit is contained in:
Zachary Yedidia
2016-08-11 16:59:06 -04:00
parent 7c13dee33d
commit 096a403d52

View File

@@ -119,8 +119,9 @@ func (s *SplitTree) Cleanup() {
for i, node := range s.children {
if n, ok := node.(*SplitTree); ok {
if len(n.children) == 1 {
if _, ok := n.children[0].(*LeafNode); ok {
s.children[i] = n.children[0]
if child, ok := n.children[0].(*LeafNode); ok {
s.children[i] = child
child.parent = s
continue
}
}