From 7bc8e8fd722c2cb3ceb735451c6dbb55f421057b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?BU=20Fanchen=20=E5=8D=9C=E5=87=A1=E8=BE=B0?=
 <boqvezen97@kaist.ac.kr>
Date: Sat, 11 May 2024 11:58:07 +0900
Subject: [PATCH] Update sbm_estimators.py

bug fix for size-one blocks with loops=False
---
 graspologic/models/sbm_estimators.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/graspologic/models/sbm_estimators.py b/graspologic/models/sbm_estimators.py
index 1787662d5..05793619a 100644
--- a/graspologic/models/sbm_estimators.py
+++ b/graspologic/models/sbm_estimators.py
@@ -516,6 +516,8 @@ def _calculate_block_p(
         if return_counts:
             p = np.count_nonzero(block)
         else:
+            if block.size == 0:
+                continue
             p = _calculate_p(block)
         block_p[from_block, to_block] = p
     return block_p