Skip to content

Commit e3cde04

Browse files
authored
Update dist_neighbor_loader.py
1 parent 716eccc commit e3cde04

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

torch_geometric/distributed/dist_neighbor_loader.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,10 @@
1212
from torch_geometric.loader import NodeLoader
1313
from torch_geometric.sampler.base import SubgraphType
1414
from torch_geometric.typing import EdgeType, InputNodes, OptTensor
15+
from torch_geometric.deprecation import deprecated
1516

16-
17+
@deprecated("use 'CuGraph' instead\
18+
https://github.com/rapidsai/cugraph")
1719
class DistNeighborLoader(NodeLoader, DistLoader):
1820
r"""A distributed loader that performs sampling from nodes.
1921

0 commit comments

Comments
 (0)