From 1b1073fe7259fea1167a069200b0d8693fb474e5 Mon Sep 17 00:00:00 2001 From: tessier Date: Sat, 12 Feb 2022 23:32:09 +0900 Subject: [PATCH] refactor: Reflect changes in Cluster class --- diagrams/__init__.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/diagrams/__init__.py b/diagrams/__init__.py index 1fb33db7..33520175 100644 --- a/diagrams/__init__.py +++ b/diagrams/__init__.py @@ -258,11 +258,8 @@ class Cluster: self._diagram.subgraph(self.dot) setcluster(self._parent) - def _validate_direction(self, direction: str): - direction = direction.upper() - if direction in self.__directions: - return True - return False + def _validate_direction(self, direction: str) -> bool: + return direction.upper() in self.__directions def node(self, nodeid: str, label: str, **attrs) -> None: """Create a new node in the cluster."""