refactor: Reflect changes in Cluster class (mingrammer#653)

aneoconsulting · Nov 25, 2022 · 09944ab · 09944ab
1 parent e150842
commit 09944ab
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/diagrams/__init__.py b/diagrams/__init__.py
@@ -260,11 +260,8 @@ def __exit__(self, exc_type, exc_value, traceback):
             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."""