From f327fcb498c7d4203dfc7e6626b2a4e3e9142813 Mon Sep 17 00:00:00 2001
From: Tiago de Paula Peixoto <tiago@skewed.de>
Date: Fri, 17 Feb 2017 14:07:37 +0000
Subject: [PATCH] generate_sbm(): fix bug when B=1

---
 src/graph_tool/generation/__init__.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/graph_tool/generation/__init__.py b/src/graph_tool/generation/__init__.py
index ef4bdd90..02fe7d71 100644
--- a/src/graph_tool/generation/__init__.py
+++ b/src/graph_tool/generation/__init__.py
@@ -978,6 +978,9 @@ def generate_sbm(b, probs, out_degs=None, in_degs=None, directed=False):
         s = s[idx]
     p = numpy.squeeze(numpy.array(probs[r, s]))
 
+    if len(p.shape) == 0: # B == 1 special case
+        p = numpy.array([p])
+
     g.set_directed(directed)
 
     libgraph_tool_generation.gen_sbm(g._Graph__graph,
-- 
GitLab