Skip to content

Commit

Permalink
fix typos
Browse files Browse the repository at this point in the history
  • Loading branch information
Ubuntu committed Dec 13, 2024
1 parent d8cb4b4 commit 0f347e3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/network_traffic_prediction/nr_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -363,7 +363,7 @@ def forward(self, blocks, node_feats, edge_feats, labels, input_nodes=None):
Input node features for each node type in the format of {ntype: tensor}.
edge_feats: list of dict of tensors
Input edge features for each edge type in the format of [{etype: tensor}, ...],
or [{}, {}. ...] for zero number of edges in input blocks. The length of e_hs
or [{}, {}. ...] for zero number of edges in input blocks. The length of edge_feats
should be equal to the number of gnn layers.
labels: dict of tensor
A dict of node/edge type and lables in format {type_str, tensor}.
Expand Down

0 comments on commit 0f347e3

Please sign in to comment.