Skip to content

GitLab

"src/array/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "183d29de56f0115e9e3de7f3b8c1966c2e6a5470"
Unverified Commit d8c69d53 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

Fixing typo in JTNN after interface change (#536)

parent acefb9a9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/jtnn/jtnn/jtnn_enc.py
View file @ d8c69d53
...@@ -15,7 +15,7 @@ MAX_NB = 8 ...@@ -15,7 +15,7 @@ MAX_NB = 8
def level_order(forest, roots): def level_order(forest, roots):
edges = bfs_edges_generator(forest, roots) edges = bfs_edges_generator(forest, roots)
_, leaves = forest.find_edges(edges[-1]) _, leaves = forest.find_edges(edges[-1])
edges_back = bfs_edges_generator(forest, roots, reversed=True) edges_back = bfs_edges_generator(forest, roots, reverse=True)
yield from reversed(edges_back) yield from reversed(edges_back)
yield from edges yield from edges
......
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