Fix deprecation warnings

codingjoe · Nov 22, 2022 · 5933bd1 · 5933bd1
1 parent 1385831
commit 5933bd1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/joeflow/models.py b/joeflow/models.py
@@ -197,7 +197,7 @@ def get_graph(cls, color="black"):
         graph.attr("graph", rankdir="LR")
         graph.attr(
             "node",
-            dict(
+            _attributes=dict(
                 fontname="sans-serif", shape="rect", style="filled", fillcolor="white"
             ),
         )

diff --git a/joeflow/utils.py b/joeflow/utils.py
@@ -35,15 +35,15 @@ def __iter__(self, subgraph=False):
         for kw in ("graph", "node", "edge"):
             attrs = getattr(self, "%s_attr" % kw)
             if attrs:
-                yield self._attr(kw, self._attr_list(None, attrs))
+                yield self._attr(kw, self._attr_list(None, kwargs=attrs))
 
         yield from self.body
 
         for name, attrs in sorted(self._nodes.items()):
             name = self._quote(name)
             label = attrs.pop("label", None)
             _attributes = attrs.pop("_attributes", None)
-            attr_list = self._attr_list(label, attrs, _attributes)
+            attr_list = self._attr_list(label, kwargs=attrs, attributes=_attributes)
             yield self._node(name, attr_list)
 
         for edge, attrs in sorted(self._edges.items()):
@@ -52,7 +52,7 @@ def __iter__(self, subgraph=False):
             head_name = self._quote_edge(head_name)
             label = attrs.pop("label", None)
             _attributes = attrs.pop("_attributes", None)
-            attr_list = self._attr_list(label, attrs, _attributes)
+            attr_list = self._attr_list(label, kwargs=attrs, attributes=_attributes)
             yield self._edge(tail=tail_name, head=head_name, attr=attr_list)
 
         yield self._tail