Skip to content

GitLab

"...text-generation-inference.git" did not exist on "9e2fdf57c04bae65827b2b03ad2b696eb6e8dec7"
Unverified Commit 75e89a15 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

remove debug information (#2056)

parent 9da1629c
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
python/dgl/sampling/neighbor.py
View file @ 75e89a15
...@@ -162,12 +162,10 @@ def sample_neighbors(g, nodes, fanout, edge_dir='in', prob=None, replace=False, ...@@ -162,12 +162,10 @@ def sample_neighbors(g, nodes, fanout, edge_dir='in', prob=None, replace=False,
# only set the edge IDs. # only set the edge IDs.
if not _dist_training: if not _dist_training:
if copy_ndata: if copy_ndata:
print(g, type(g))
node_frames = utils.extract_node_subframes(g, None) node_frames = utils.extract_node_subframes(g, None)
utils.set_new_frames(ret, node_frames=node_frames) utils.set_new_frames(ret, node_frames=node_frames)
if copy_edata: if copy_edata:
print(g, type(g))
edge_frames = utils.extract_edge_subframes(g, induced_edges) edge_frames = utils.extract_edge_subframes(g, induced_edges)
utils.set_new_frames(ret, edge_frames=edge_frames) utils.set_new_frames(ret, edge_frames=edge_frames)
else: else:
...@@ -296,12 +294,10 @@ def select_topk(g, k, weight, nodes=None, edge_dir='in', ascending=False, ...@@ -296,12 +294,10 @@ def select_topk(g, k, weight, nodes=None, edge_dir='in', ascending=False,
# handle features # handle features
if copy_ndata: if copy_ndata:
print(g, type(g))
node_frames = utils.extract_node_subframes(g, None) node_frames = utils.extract_node_subframes(g, None)
utils.set_new_frames(ret, node_frames=node_frames) utils.set_new_frames(ret, node_frames=node_frames)
if copy_edata: if copy_edata:
print(g, type(g))
edge_frames = utils.extract_edge_subframes(g, induced_edges) edge_frames = utils.extract_edge_subframes(g, induced_edges)
utils.set_new_frames(ret, edge_frames=edge_frames) utils.set_new_frames(ret, edge_frames=edge_frames)
return ret return ret
......
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