From d0c803cb7dedf459797946a8e6687d8fffbd00d9 Mon Sep 17 00:00:00 2001
From: Jeffrey Wigger <jeffrey.wigger@epfl.ch>
Date: Mon, 13 Jun 2022 18:20:05 +0200
Subject: [PATCH] neighbor fixes 3

---
 eval/run_xtimes_reddit_rws.sh                            | 2 +-
 src/decentralizepy/communication/TCPRandomWalkRouting.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/eval/run_xtimes_reddit_rws.sh b/eval/run_xtimes_reddit_rws.sh
index b6024de..c206222 100755
--- a/eval/run_xtimes_reddit_rws.sh
+++ b/eval/run_xtimes_reddit_rws.sh
@@ -71,7 +71,7 @@ echo samples per user: $samples_per_user
 
 # random_seeds for which to rerun the experiments
 # random_seeds=("90" "91" "92" "93" "94")
-random_seeds=("90" "91" "92")
+random_seeds=("92")
 echo batchsize: $batchsize
 echo communication rounds per global epoch: $comm_rounds_per_global_epoch
 # calculating how many batches there are in a global epoch for each user/proc
diff --git a/src/decentralizepy/communication/TCPRandomWalkRouting.py b/src/decentralizepy/communication/TCPRandomWalkRouting.py
index fd5f638..1433420 100644
--- a/src/decentralizepy/communication/TCPRandomWalkRouting.py
+++ b/src/decentralizepy/communication/TCPRandomWalkRouting.py
@@ -427,6 +427,7 @@ class TCPRandomWalkRoutingInternal(TCPRandomWalkBase):
             if src in self.current_neighbors:
                 logging.info("fw arrived at a neighbour")
                 print("fw arrived at a neighbour")
+                self.add_neighbors(src, data[1])
             else:
                 logging.debug("Received {} from {}".format(data, src))
                 self.add_neighbors(src, data[1])
-- 
GitLab