Fix initialization of overlapping NestedBlockState

This fixes issue #678

Fix initialization of overlapping NestedBlockState
This fixes issue #678
dbc8e6f4 · Tiago Peixoto · 711f2f3d · dbc8e6f4
Commit dbc8e6f4 authored Sep 15, 2020 by Tiago Peixoto
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/graph_tool/inference/nested_blockmodel.py src/graph_tool/inference/nested_blockmodel.py +1 -1

No files found.
--- a/src/graph_tool/inference/nested_blockmodel.py
+++ b/src/graph_tool/inference/nested_blockmodel.py
@@ -66,7 +66,7 @@ class NestedBlockState(object):

        if bs is None:
            if base_type is OverlapBlockState or state_args.get("overlap", False):
-                b = zeros(2 * g.num_vertices(), dtype="int")
+                b = zeros(2 * g.num_edges(), dtype="int")
            else:
                b = zeros(g.num_vertices(), dtype="int")
            bs = [b] + [zeros(1, dtype="int")] * int(ceil(log2(len(b))))