From 35ca173b2ef421f41fac44cd4100c1db865c28f0 Mon Sep 17 00:00:00 2001
From: Nick Hoo <442869088@qq.com>
Date: Mon, 20 Sep 2021 12:37:29 -0400
Subject: [PATCH] Update preprocessing.py

Do we need to repeat this sentence?
if ismember([idx_j, idx_i], np.array(val_edges_false)):
                continue
---
 preprocessing.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/preprocessing.py b/preprocessing.py
index 6ee6fc7..9c28c2b 100644
--- a/preprocessing.py
+++ b/preprocessing.py
@@ -86,8 +86,6 @@ def ismember(a, b, tol=5):
         if val_edges_false:
             if ismember([idx_j, idx_i], np.array(val_edges_false)):
                 continue
-            if ismember([idx_i, idx_j], np.array(val_edges_false)):
-                continue
         val_edges_false.append([idx_i, idx_j])
 
     assert ~ismember(test_edges_false, edges_all)
@@ -103,4 +101,4 @@ def ismember(a, b, tol=5):
     adj_train = adj_train + adj_train.T
 
     # NOTE: these edge lists only contain single direction of edge!
-    return adj_train, train_edges, val_edges, val_edges_false, test_edges, test_edges_false
\ No newline at end of file
+    return adj_train, train_edges, val_edges, val_edges_false, test_edges, test_edges_false