From 053bf2cb461e47fc6438ab545b2ba493149954c1 Mon Sep 17 00:00:00 2001 From: Attila Nagy Date: Tue, 19 Jan 2021 14:47:38 +0100 Subject: [PATCH] Allow Node arguments to be passed to Custom Node --- diagrams/custom/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/diagrams/custom/__init__.py b/diagrams/custom/__init__.py index 48441a6e..9845932d 100644 --- 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)