From f737271303aa3a06194e1ec0f5b51262f410f19a Mon Sep 17 00:00:00 2001
From: HiccupHan <89772977+HiccupHan@users.noreply.github.com>
Date: Mon, 3 Mar 2025 22:19:49 -0800
Subject: [PATCH] fix for sampling issue
---
app/database.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/app/database.py b/app/database.py
index aa55595..41f5b74 100644
--- a/app/database.py
+++ b/app/database.py
@@ -24,7 +24,8 @@ def get_friends(user_id, num_friends):
random.seed(0)
if not curr_user:
return {}
- for f in random.sample(curr_user[0]["friends"], num_friends):
+ sample_size = min(num_friends, len(curr_user[0]["friends"]))
+ for f in random.sample(curr_user[0]["friends"], sample_size):
friends[f] = db.search(User.user_id == user_id)[0]
return friends