✨ fix: fix typing bug

mmsaki · Feb 4, 2025 · b49b6d3 · b49b6d3
1 parent 69ef2ee
commit b49b6d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/defigraph/Graph.py b/src/defigraph/Graph.py
@@ -25,10 +25,10 @@ def __init__(self, edges: List[Edge]) -> None:
     def __repr__(self) -> str:
         return f"{self.adjascency_list}"
 
-    def __getitem__(self, vertex: str) -> List[Edge]:
+    def __getitem__(self, vertex: Vertex) -> List[Edge]:
         return self.adjascency_list[vertex]
 
-    def __setitem__(self, vertex: str, edges: List[Vertex]) -> None:
+    def __setitem__(self, vertex: Vertex, edges: List[Edge]) -> None:
         if vertex not in self.adjascency_list:
             self.adjascency_list[vertex] = edges
         else: