Skip to content

Commit

Permalink
Fixed typo in NegativeSampling docstring (#9533)
Browse files Browse the repository at this point in the history
docstring for torch_geometric.sampler.base.NegativeSampling said
src_weight for the parameter dst_weight (issue #9512)
  • Loading branch information
bryceForrest authored Jul 24, 2024
1 parent 6037320 commit 2fdee7f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch_geometric/sampler/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -537,7 +537,7 @@ class NegativeSampling(CastMixin):
the sampling of source nodes. Does not necessarily need to sum up
to one. If not given, negative nodes will be sampled uniformly.
(default: :obj:`None`)
src_weight (torch.Tensor, optional): A node-level vector determining
dst_weight (torch.Tensor, optional): A node-level vector determining
the sampling of destination nodes. Does not necessarily need to sum
up to one. If not given, negative nodes will be sampled uniformly.
(default: :obj:`None`)
Expand Down

0 comments on commit 2fdee7f

Please sign in to comment.