Refactor Graph class to use __getstate__ and

__setstate__ methods
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-27 18:29:44 -03:00
commit 81279853a0

View file

@ -34,11 +34,11 @@ class Graph:
self._edges = self._graph_data["edges"]
self._build_graph()
def __getstate__(self):
return self.raw_graph_data
def __setstate__(self, state):
self.__dict__.update(state)
for edge in self.edges:
edge.reset()
edge.validate_edge()
self.__init__(**state)
@classmethod
def from_payload(cls, payload: Dict) -> "Graph":