Commit 4d95c40b authored by Tiago Peixoto's avatar Tiago Peixoto

Fix bug in NestedBlockState.get_edges_prob() when sampling=True

parent f8b763a5
......@@ -320,6 +320,7 @@ class NestedBlockState(object):
lstate._couple_state(None, None)
if l > 0:
lstate._state.sync_emat()
lstate._state.clear_egroups()
L += lstate.get_edges_prob(edge_list, missing, entropy_args=eargs)
if isinstance(self.levels[0], LayeredBlockState):
......
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