Fix ref assignment in render (#2901)

chaiNNer-org · May 22, 2024 · ba510a9 · ba510a9
1 parent 1038d4b
commit ba510a9
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/renderer/components/ReactFlowBox.tsx b/src/renderer/components/ReactFlowBox.tsx
@@ -86,8 +86,10 @@ export const ReactFlowBox = memo(({ wrapperRef, nodeTypes, edgeTypes }: ReactFlo
 
     const [nodes, setNodes, internalOnNodesChange] = useNodesState<NodeData>([]);
     const [edges, setEdges, internalOnEdgesChange] = useEdgesState<EdgeData>([]);
-    setNodesRef.current = setNodes;
-    setEdgesRef.current = setEdges;
+    useEffect(() => {
+        setNodesRef.current = setNodes;
+        setEdgesRef.current = setEdges;
+    }, [setNodes, setEdges, setNodesRef, setEdgesRef]);
 
     const altPressed = useKeyPress(['Alt', 'Option']);