Browse Source

refactor: Reflect changes in Cluster class (#653)

pull/650/merge
gabriel-tessier 2 years ago
committed by GitHub
parent
commit
eeb1dc7f2b
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
  1. +2
    -5
      diagrams/__init__.py

+ 2
- 5
diagrams/__init__.py View File

@@ -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):
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: 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