refactor: Reflect changes in Cluster class (#653)

pull/650/merge
gabriel-tessier 2 years ago committed by GitHub
parent 82b56e6edd
commit eeb1dc7f2b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -260,11 +260,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