fix topology merging

99b5bf01a24a2b5960224e9c86c215542f309f86 · 2024-12-05 21:12:47 +0000 · Alex Cheema

Files touched

M exo/topology/topology.py

Diff

commit 99b5bf01a24a2b5960224e9c86c215542f309f86
Author: Alex Cheema <alexcheema123@gmail.com>
Date:   Thu Dec 5 21:12:47 2024 +0000

    fix topology merging
---
 exo/topology/topology.py | 17 +++++------------
 1 file changed, 5 insertions(+), 12 deletions(-)

diff --git a/exo/topology/topology.py b/exo/topology/topology.py
index 2ec5a2ab..5549a9cb 100644
--- a/exo/topology/topology.py
+++ b/exo/topology/topology.py
@@ -34,18 +34,11 @@ class Topology:
   def all_nodes(self):
     return self.nodes.items()
 
-  def add_edge(self, node1_id: str, node2_id: str, description: Optional[str] = None):
-    if node1_id not in self.peer_graph:
-      self.peer_graph[node1_id] = set()
-    if node2_id not in self.peer_graph:
-      self.peer_graph[node2_id] = set()
-
-    # Create bidirectional connections with the same description
-    conn1 = PeerConnection(node1_id, node2_id, description)
-    conn2 = PeerConnection(node2_id, node1_id, description)
-
-    self.peer_graph[node1_id].add(conn1)
-    self.peer_graph[node2_id].add(conn2)
+  def add_edge(self, from_id: str, to_id: str, description: Optional[str] = None):
+    if from_id not in self.peer_graph:
+      self.peer_graph[from_id] = set()
+    conn = PeerConnection(from_id, to_id, description)
+    self.peer_graph[from_id].add(conn)
 
   def get_neighbors(self, node_id: str) -> Set[str]:
     # Convert PeerConnection objects back to just destination IDs

← d5b4039f show interface type and name, fix propagation of topology · back to Exo · handle mutable visited properly dba72044 →