Add predecessors and successors properties to Vertex class

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-21 09:57:49 -03:00
commit d35a3d7b87

View file

@ -113,6 +113,14 @@ class Vertex:
def edges(self) -> List["ContractEdge"]:
return self.graph.get_vertex_edges(self.id)
@property
def predecessors(self) -> List["Vertex"]:
return self.graph.get_predecessors(self)
@property
def successors(self) -> List["Vertex"]:
return self.graph.get_successors(self)
def __getstate__(self):
return {
"_data": self._data,