Skip to content

GitLab

Commit 16da76c4 authored by Gan Quan's avatar Gan Quan
Browse files

fix #81

parent 5567f4a4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/graph.py
View file @ 16da76c4
...@@ -884,7 +884,7 @@ class DGLGraph(object): ...@@ -884,7 +884,7 @@ class DGLGraph(object):
def _batch_update_edge(self, u, v, edge_func): def _batch_update_edge(self, u, v, edge_func):
if is_all(u) and is_all(v): if is_all(u) and is_all(v):
u, v = self._graph.edges() u, v, _ = self._graph.edges()
# call the UDF # call the UDF
src_reprs = self.get_n_repr(u) src_reprs = self.get_n_repr(u)
dst_reprs = self.get_n_repr(v) dst_reprs = self.get_n_repr(v)
......
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