Pathfinding: Added a small fix (ugly but works)

Piratmac · Piratmac · commit bb7926951671 · 2020-08-24T21:40:11.000+02:00
diff --git a/2019/pathfinding.py b/2019/pathfinding.py
@@ -513,7 +513,12 @@ def dijkstra(self, start, end=None):
                     continue
 
                 # Adding for future examination
-                heapq.heappush(frontier, (current_distance + weight, neighbor))
+                if type(neighbor) == complex:
+                    heapq.heappush(
+                        frontier, (current_distance + weight, SuperComplex(neighbor))
+                    )
+                else:
+                    heapq.heappush(frontier, (current_distance + weight, neighbor))
 
                 # Adding for final search
                 self.distance_from_start[neighbor] = current_distance + weight
