Commit 4a4cebb7 authored by Tiago Peixoto's avatar Tiago Peixoto
Browse files

Cosmetic blockmodel.py fixes

parent ee978e1d
......@@ -2923,9 +2923,8 @@ void move_sweep(vector<BlockState>& states, vector<MEntries>& m_entries_r,
vector<std::tuple<vertex_t, vertex_t, double> >,
decltype(merge_cmp_less)> move_heap(merge_cmp_less);
for (size_t i = 0; i < num_vertices(g); ++i)
for (vertex_t v : vlist)
{
vertex_t v = vertex(i, g);
vertex_t r = b[v];
vertex_t s = best_move[v].first;
double dS = best_move[v].second;
......
......@@ -1313,8 +1313,8 @@ def greedy_shrink(state, B, **kwargs):
# merge according to indirect neighbourhood; we put all group-nodes in their
# own groups, and merge/move them until the desired size is reached
curr_B = (state.wr.a > 0).sum()
assert curr_B >= B, "shrinking to a larger size ?! (%d, %d)" % (curr_B, B)
curr_B = (state.wr.fa > 0).sum()
assert curr_B > B, "shrinking to a larger size ?! (%d, %d, %d)" % (state.B, curr_B, B)
random = kwargs.get("random_move", False)
old_state = state
......
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