remove tag and parent named params

hbmartin · Jul 2, 2024 · 99ad032 · 99ad032
1 parent 57d2ea2
commit 99ad032
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/graphviz2drawio/mx/MxGraph.py b/graphviz2drawio/mx/MxGraph.py
@@ -11,10 +11,8 @@ def __init__(self, nodes, edges):
         self.edges = edges
         self.graph = ET.Element(MxConst.GRAPH)
         self.root = ET.SubElement(self.graph, MxConst.ROOT)
-        ET.SubElement(parent=self.root, tag=MxConst.CELL, attrib={"id": "0"})
-        ET.SubElement(
-            parent=self.root, tag=MxConst.CELL, attrib={"id": "1", "parent": "0"}
-        )
+        ET.SubElement(self.root, MxConst.CELL, attrib={"id": "0"})
+        ET.SubElement(self.root, MxConst.CELL, attrib={"id": "1", "parent": "0"})
 
         for edge in edges:
             self.add_edge(edge)
@@ -25,8 +23,8 @@ def add_edge(self, edge):
         source, target = self.get_edge_source_target(edge)
         style = self.get_edge_style(edge, source, target)
         edge_element = ET.SubElement(
-            parent=self.root,
-            tag=MxConst.CELL,
+            self.root,
+            MxConst.CELL,
             attrib={
                 "id": edge.sid,
                 "style": style,
@@ -90,8 +88,8 @@ def add_node(self, node):
         style = Styles.get_for_shape(node.shape).format(fill=fill, stroke=stroke)
 
         node_element = ET.SubElement(
-            parent=self.root,
-            tag=MxConst.CELL,
+            self.root,
+            MxConst.CELL,
             attrib={
                 "id": node.sid,
                 "value": node.text_to_mx_value(),