Commit 82e496e6 authored by Tiago Peixoto's avatar Tiago Peixoto
Browse files

Fix issue with edge prediction in layered models

parent 9dd4ae83
Pipeline #190 failed with stage
in 11012 minutes and 18 seconds
......@@ -657,7 +657,7 @@ class LayeredBlockState(OverlapBlockState, BlockState):
try:
u, v = e
l = self.ec[e]
except ValueError:
except (TypeError, ValueError):
u, v, l = e
pos[u] = self.b[u]
......
......@@ -302,7 +302,7 @@ class NestedBlockState(object):
eargs = overlay(eargs, **self.hentropy_args)
L += state.get_edges_prob(edge_list, missing=missing,
entropy_args=eargs)
edge_list = [(state.b[u], state.b[v]) for u, v in edge_list]
edge_list = [(state.b[e[0]], state.b[e[1]]) for e in (tuple(e_) for e_ in edge_list)]
return L
def get_bstack(self):
......
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