Merge pull request #40 from Luke-Poeppel/37_remove_child_bugfix

Fixes #37. Bugfix for `remove_child` (Credit: Ricardo Reis)
Luke-Poeppel · Sep 20, 2021 · e254c35 · e254c35
2 parents e912f90 + b9a108b
commit e254c35
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/treeplotter/tree.py b/treeplotter/tree.py
@@ -104,13 +104,13 @@ def add_path_of_children(self, path, final_node_name):
 		return
 
 	def remove_child(self, child_node):
-		if child_node.item.value not in self.children:
+		if child_node not in self.children:
 			raise NodeException("This parent does not have that child.")
-		self.children.remove(child_node.item.value)
+		self.children.remove(child_node)
 		return
 
 	def remove_children(self, children_nodes):
-		for this_child in children_nodes:
+		for this_child in tuple(children_nodes):
 			try:
 				self.remove_child(this_child)
 			except KeyError: