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