Skip to content

GitLab

Commit 9cc83b4b authored by Pin's avatar Pin Committed by Minjie Wang
Browse files

[Bugfix] Fix nodeflow.prop_flow range check (#700)

parent bf3994ee
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/nodeflow.py
View file @ 9cc83b4b
...@@ -938,7 +938,7 @@ class NodeFlow(DGLBaseGraph): ...@@ -938,7 +938,7 @@ class NodeFlow(DGLBaseGraph):
if is_all(flow_range): if is_all(flow_range):
flow_range = range(0, self.num_blocks) flow_range = range(0, self.num_blocks)
elif isinstance(flow_range, slice): elif isinstance(flow_range, slice):
if slice.step != 1: if flow_range.step != 1:
raise DGLError("We can't propogate flows and skip some of them") raise DGLError("We can't propogate flows and skip some of them")
flow_range = range(flow_range.start, flow_range.stop) flow_range = range(flow_range.start, flow_range.stop)
else: else:
......
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