Skip to content

Commit 4497a62

Browse files
committed
fix typos
1 parent 2dda663 commit 4497a62

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

models/MemoryModel.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -553,7 +553,7 @@ def __init__(self, node_raw_features: torch.Tensor, edge_raw_features: torch.Ten
553553
"""
554554
Graph attention embedding module.
555555
:param node_raw_features: Tensor, shape (num_nodes + 1, node_feat_dim)
556-
:param edge_raw_features: Tensor, shape (num_edges + 1, node_feat_dim)
556+
:param edge_raw_features: Tensor, shape (num_edges + 1, edge_feat_dim)
557557
:param neighbor_sampler: NeighborSampler, neighbor sampler
558558
:param time_encoder: TimeEncoder
559559
:param node_feat_dim: int, dimension of node features

0 commit comments

Comments
 (0)