From 475600738040cbde480395daf645c597edf85384 Mon Sep 17 00:00:00 2001
From: Aaron <aaron@hiddenharvestventures.com>
Date: Mon, 26 May 2025 15:56:49 -0400
Subject: [PATCH] Handle choosing a filepair when there are less than 10 to
 choose from

---
 ridges/validator/generate_problem.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/ridges/validator/generate_problem.py b/ridges/validator/generate_problem.py
index bd73e31..29942d3 100644
--- a/ridges/validator/generate_problem.py
+++ b/ridges/validator/generate_problem.py
@@ -62,11 +62,16 @@ class ListOfGeneratedProblems(BaseModel):
 
 
 def highest_cosine_filepair_selector(file_pairs: List[FilePair]) -> FilePair:
-    selected_file_pair = sorted(
+    if len(file_pairs) < 1:
+        raise ValueError("No file pairs found")
+
+    top_ten_file_pairs = sorted(
         file_pairs,
         key=lambda x: float(x.cosine_similarity),
         reverse=True
-    )[random.randint(1, 10)]
+    )[:10]
+
+    selected_file_pair = random.choice(top_ten_file_pairs)
 
     return selected_file_pair