feat: allow Node arguments to be passed to Custom Node (mingrammer#437)

aneoconsulting · Nov 25, 2022 · ec11a31 · ec11a31
1 parent 106411a
commit ec11a31
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/diagrams/custom/__init__.py b/diagrams/custom/__init__.py
@@ -15,6 +15,6 @@ class Custom(Node):
     def _load_icon(self):
         return self._icon
 
-    def __init__(self, label, icon_path):
+    def __init__(self, label, icon_path, *args, **kwargs):
         self._icon = icon_path
-        super().__init__(label)
+        super().__init__(label, *args, **kwargs)