jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor Graph class to use __getstate__ and

Browse source 
__setstate__ methods
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-27 18:29:44 -03:00
commit 81279853a0
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/backend/langflow/graph/graph/base.py
View file

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