refactor: Reflect changes in Cluster class

pull/653/head
tessier 4 years ago
parent eaf3e98304
commit 1b1073fe72

@ -258,11 +258,8 @@ class Cluster:
self._diagram.subgraph(self.dot) self._diagram.subgraph(self.dot)
setcluster(self._parent) setcluster(self._parent)
def _validate_direction(self, direction: str): def _validate_direction(self, direction: str) -> bool:
direction = direction.upper() return direction.upper() in self.__directions
if direction in self.__directions:
return True
return False
def node(self, nodeid: str, label: str, **attrs) -> None: def node(self, nodeid: str, label: str, **attrs) -> None:
"""Create a new node in the cluster.""" """Create a new node in the cluster."""

Loading…
Cancel
Save