From 70d1c8e3b0d1662a4c39aada8c6b13b8b50a8338 Mon Sep 17 00:00:00 2001
From: Matthew Bryan <matthewbryan52@gmail.com>
Date: Tue, 30 Sep 2025 12:13:52 +0200
Subject: [PATCH] Fix `random_coords`, at least 1 coord returned

---
 src/temgym_core/utils.py | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/src/temgym_core/utils.py b/src/temgym_core/utils.py
index f7cbb0a..20200d4 100755
--- a/src/temgym_core/utils.py
+++ b/src/temgym_core/utils.py
@@ -175,7 +175,10 @@ def concentric_rings(
 
 
 def random_coords(num: int) -> np.ndarray:
-    """Draw uniformly random points within a unit-radius disc.
+    """
+    Draw uniformly random points within a unit-radius disc.
+
+    Guarantees that at least one coordinate will be returned.
 
     Parameters
     ----------
@@ -187,13 +190,16 @@ def random_coords(num: int) -> np.ndarray:
     yx : numpy.ndarray, shape (N, 2), float64
         Coordinates (y, x) in [-1, 1] within the unit disc.
     """
-    yx = np.random.uniform(
-        -1,
-        1,
-        size=(int(num * 1.28), 2),  # 4 / np.pi
-    )
-    radii = np.sqrt((yx**2).sum(axis=1))
-    mask = radii < 1
+    while True:
+        yx = np.random.uniform(
+            -1,
+            1,
+            size=(max(1, int(num * 1.28)), 2),  # 4 / np.pi
+        )
+        radii = np.sqrt((yx**2).sum(axis=1))
+        mask = radii < 1
+        if mask.sum() > 0:
+            break
     yx = yx[mask, :]
     return yx