Edited name of function

GreeleyGroup · Sep 7, 2023 · 6f5e207 · 6f5e207
1 parent fa0f476
commit 6f5e207
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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