Unverified Commit 101d2ae9 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Misc] Rename has_origin_eids to has_original_eids to keep consistency. (#6446)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent d2eca855
......@@ -498,12 +498,17 @@ class CSCSamplingGraph(SamplingGraph):
"""
# Ensure nodes is 1-D tensor.
self._check_sampler_arguments(nodes, fanouts, probs_name)
has_origin_eids = (
has_original_eids = (
self.edge_attributes is not None
and ORIGINAL_EDGE_ID in self.edge_attributes
)
return self._c_csc_graph.sample_neighbors(
nodes, fanouts.tolist(), replace, False, has_origin_eids, probs_name
nodes,
fanouts.tolist(),
replace,
False,
has_original_eids,
probs_name,
)
def 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