From d250a7d2cdb544b4f4ed527347ea6fb88076e671 Mon Sep 17 00:00:00 2001
From: himself65 <himself65@outlook.com>
Date: Sun, 6 Feb 2022 17:11:50 -0600
Subject: [PATCH] fix: enhance type

---
 src/utils/graph.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/utils/graph.ts b/src/utils/graph.ts
index e6457789..fc7b1f32 100644
--- a/src/utils/graph.ts
+++ b/src/utils/graph.ts
@@ -9,7 +9,7 @@ export const isEdge = (element: Node | Connection | Edge): element is Edge =>
 export const isNode = (element: Node | Connection | Edge): element is Node =>
   'id' in element && !('source' in element) && !('target' in element);
 
-export const getOutgoers = (node: Node, nodes: Node[], edges: Edge[]): Node[] => {
+export const getOutgoers = <T = any, U extends T = T>(node: Node<U>, nodes: Node<T>[], edges: Edge[]): Node<T>[] => {
   if (!isNode(node)) {
     return [];
   }
@@ -18,7 +18,7 @@ export const getOutgoers = (node: Node, nodes: Node[], edges: Edge[]): Node[] =>
   return nodes.filter((n) => outgoerIds.includes(n.id));
 };
 
-export const getIncomers = (node: Node, nodes: Node[], edges: Edge[]): Node[] => {
+export const getIncomers = <T = any, U extends T = T>(node: Node<U>, nodes: Node<T>[], edges: Edge[]): Node<T>[] => {
   if (!isNode(node)) {
     return [];
   }