debug

pyg-team · Jan 10, 2025 · 1ca06cd · 1ca06cd
1 parent 3a7d7b2
commit 1ca06cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_geometric/data/large_graph_indexer.py b/torch_geometric/data/large_graph_indexer.py
@@ -540,7 +540,7 @@ def to_data(self, node_feature_name: str,
         print("self.get_edge_features(EDGE_INDEX)[:10]=",self.get_edge_features(EDGE_INDEX)[:10])
         edge_index = torch.t(
             torch.LongTensor(self.get_edge_features(EDGE_INDEX)))
-        print("edge_index tensor of full graph=", edge_index)
+        print("edge_index tensor of full graph=", edge_index[:, :10])
 
         edge_attr = (self.get_edge_features(edge_feature_name)
                      if edge_feature_name is not None else None)