From 55f012b514fcfaf6cdbcbf7c9b8709574a53f155 Mon Sep 17 00:00:00 2001 From: anovis Date: Mon, 23 Aug 2021 16:13:17 -0400 Subject: [PATCH] update based on review --- diagrams/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/diagrams/__init__.py b/diagrams/__init__.py index 2a388f5f..4de1a527 100644 --- a/diagrams/__init__.py +++ b/diagrams/__init__.py @@ -304,8 +304,11 @@ class Node: raise EnvironmentError("Global diagrams context not set up") if self._diagram.autolabel: - # import pdb; pdb.set_trace() - self.label = self.__class__.__name__ + "\n" + self.label + prefix = self.__class__.__name__ + if self.label: + self.label = prefix + "\n" + self.label + else: + self.label = prefix # fmt: off # If a node has an icon, increase the height slightly to avoid