From 6f5e207ce2378d31a6da6fae59d1658b0c6c9eda Mon Sep 17 00:00:00 2001 From: Gaurav S Deshmukh Date: Thu, 7 Sep 2023 17:13:40 -0400 Subject: [PATCH] Edited name of function --- src/graphs.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/graphs.py b/src/graphs.py index 76c8048..fdead15 100644 --- a/src/graphs.py +++ b/src/graphs.py @@ -107,13 +107,13 @@ def get_node_tensor(self): """ pass - def get_adjacency_tensor(self): - """Get the adjacency matrix of the graph as a PyTorch Tensor. + def get_edge_tensor(self): + """Get the edge matrix of the graph as a PyTorch Tensor. Returns ------- - adj_matrix: torch.Tensor - Adjacency tensor + edge_matrix: torch.Tensor + Edge tensor """ pass