Fix docstring of GraphTensor.edge_sets: it's not for node sets.

PiperOrigin-RevId: 650864543
tensorflow · Jul 10, 2024 · 95b2ac9 · 95b2ac9
1 parent 5443fb7
commit 95b2ac9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_gnn/graph/graph_tensor.py b/tensorflow_gnn/graph/graph_tensor.py
@@ -1260,7 +1260,7 @@ def node_sets(self) -> Mapping[NodeSetName, NodeSet]:
 
   @property
   def edge_sets(self) -> Mapping[EdgeSetName, EdgeSet]:
-    """A read-only mapping from node set name to the node set."""
+    """A read-only mapping from edge set name to the edge set."""
     return _as_immutable_mapping(self._data[GraphTensor._DATAKEY_EDGE_SETS])
 
   @property