Chore: Refactor remove function type usage

memgraph · Mar 14, 2024 · 558c552 · 558c552
1 parent 5432587
commit 558c552
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/models/graph.ts b/src/models/graph.ts
@@ -279,15 +279,15 @@ export class Graph<N extends INodeBase, E extends IEdgeBase> implements IGraph<N
     const removedNodesData = this._removeNodes(nodeIds);
     const removedEdgesData = this._removeEdges(edgeIds);
 
-    const removedData = {
-      nodeIds: dedupArrays(removedNodesData.nodeIds, removedEdgesData.nodeIds),
-      edgeIds: dedupArrays(removedNodesData.edgeIds, removedEdgesData.edgeIds),
-    };
-
     this._applyEdgeOffsets();
     this._applyStyle();
 
     if (this._settings && this._settings.onRemoveData) {
+      const removedData: IGraphObjectsIds = {
+        nodeIds: dedupArrays(removedNodesData.nodeIds, removedEdgesData.nodeIds),
+        edgeIds: dedupArrays(removedNodesData.edgeIds, removedEdgesData.edgeIds),
+      };
+
       this._settings.onRemoveData(removedData);
     }
   }