bugfix

196e6a92 · Gan Quan · 10ec6e8a · 196e6a92
Commit 196e6a92 authored May 02, 2018 by Gan Quan
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

graph.py graph.py +2 -2

No files found.
--- a/graph.py
+++ b/graph.py
@@ -128,8 +128,8 @@ class DiGraph(nx.DiGraph, NN.Module):
                source = T.stack([self.node[u]['state'] for u, _ in ebunch])
                edge_tag = T.stack([self[u][v]['tag'] for u, v in ebunch])
                message = f(source, edge_tag)
-                for u, v in ebunch:
+                for i, (u, v) in enumerate(ebunch):
-                    self[u][v]['state'] = message
+                    self[u][v]['state'] = message[i]
            else:
                for u, v in ebunch:
                    self[u][v]['state'] = f(