Fix select of a single random element

freegs-plasma · Nov 13, 2023 · 74b215d · 74b215d
1 parent 256b94d
commit 74b215d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/freegs/optimiser.py b/freegs/optimiser.py
@@ -65,7 +65,9 @@ def pickUnique(N, m, e):
     inds = sorted(e)  # Sorted list of indices. Used to avoid clashes
     others = []  # The list of three agents
     for i in range(m):
-        newind = random.randint(0, N - 1 - i - len(e))
+        high = N - 1 - i - len(e)
+        newind = random.randint(0, high) if high > 0 else 0
+
         for ind in inds:
             if newind == ind:
                 newind += 1