From 8aa62c49b9094b8671d8e2608502f2820901c955 Mon Sep 17 00:00:00 2001 From: Vijay Prakash Dwivedi Date: Fri, 13 Oct 2023 19:35:17 +0800 Subject: [PATCH] minor clean --- data_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data_utils.py b/data_utils.py index 963b093..c8f95bf 100644 --- a/data_utils.py +++ b/data_utils.py @@ -87,9 +87,9 @@ def sparse_mx_to_torch_sparse_tensor(sparse_mx): def get_node_ids_for_all_seq(i): """Returns node ids for all seqs, sampled from 1/2 hop neighborhood""" # i, hop1indices, hop2indices, all_nodes_set, seq_length = args - global all_1hop_indices, all_2hop_indices, all_nodes_set, seq_length # node_ids_for_all_seq + global all_1hop_indices, all_2hop_indices, all_nodes_set, seq_length - hop1_neighbors = all_1hop_indices[i].tolist() # g.successors(i).tolist() + hop1_neighbors = all_1hop_indices[i].tolist() hop2_neighbors = all_2hop_indices[i].tolist() all_applicable_neighbors = hop1_neighbors + hop2_neighbors