minibn: random_walk stops at fixed points

colomoto · Mar 9, 2022 · 80fdcd4 · 80fdcd4
1 parent c719f65
commit 80fdcd4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/colomoto/minibn.py b/colomoto/minibn.py
@@ -688,8 +688,9 @@ def stop_condition(x):
         while True:
             i += 1
             nexts = list(self(x))
-            if nexts:
-                x = random.choice(nexts)
+            if not nexts or (len(nexts) == 1 and nexts[0] == x):
+                return
+            x = random.choice(nexts)
             yield x
             if i == steps:
                 break