Fixed issue in graph

Piratmac · Piratmac · commit 93a9d3c5c908 · 2021-12-15T09:50:17.000+01:00
diff --git a/2021/graph.py b/2021/graph.py
@@ -387,9 +387,7 @@ def dijkstra(self, start, end=None):
 
                 # Adding for future examination
                 if type(neighbor) == complex:
-                    heapq.heappush(
-                        frontier, (current_distance + weight, SuperComplex(neighbor))
-                    )
+                    heapq.heappush(frontier, (current_distance + weight, neighbor))
                 else:
                     heapq.heappush(frontier, (current_distance + weight, neighbor))
 
