diff --git a/contextgnn/nn/models/contextgnn.py b/contextgnn/nn/models/contextgnn.py index b113f26..f0f4f5c 100644 --- a/contextgnn/nn/models/contextgnn.py +++ b/contextgnn/nn/models/contextgnn.py @@ -21,7 +21,7 @@ class ContextGNN(RHSEmbeddingGNN): - r"""Implementation of HybridGNN model.""" + r"""Implementation of ContextGNN model.""" def __init__( self, data: HeteroData,