Use stable sorting in prune_inferior_points

pytorch · Sep 21, 2024 · 7312cf6 · 7312cf6
1 parent 8cfaee7
commit 7312cf6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/botorch/acquisition/utils.py b/botorch/acquisition/utils.py
@@ -344,7 +344,7 @@ def prune_inferior_points(
  idcs, counts = torch.unique(is_best, return_counts=True)
 
  if len(idcs) > max_points:
- counts, order_idcs = torch.sort(counts, descending=True)
+ counts, order_idcs = torch.sort(counts, stable=True, descending=True)
  idcs = order_idcs[:max_points]
 
  return X[idcs]