Unverified Commit 37fde7ed authored by yxy235's avatar yxy235 Committed by GitHub
Browse files

[GraphBolt] Add opt to LayerNeighborSampler. (#6695)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-0-133.us-west-2.compute.internal>
parent df3e7f1b
...@@ -240,8 +240,16 @@ class LayerNeighborSampler(NeighborSampler): ...@@ -240,8 +240,16 @@ class LayerNeighborSampler(NeighborSampler):
replace=False, replace=False,
prob_name=None, prob_name=None,
deduplicate=True, deduplicate=True,
# TODO: clean up once the migration is done.
output_cscformat=False,
): ):
super().__init__( super().__init__(
datapipe, graph, fanouts, replace, prob_name, deduplicate datapipe,
graph,
fanouts,
replace,
prob_name,
deduplicate,
output_cscformat,
) )
self.sampler = graph.sample_layer_neighbors self.sampler = graph.sample_layer_neighbors
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment