alu/cs239-caching
1
0
Fork 0
mirror of https://github.com/ltcptgeneral/cs239-caching.git synced 2025-04-01 20:33:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix for sampling issue

Browse Source
This commit is contained in:
HiccupHan 2025-03-03 22:19:49 -08:00
parent d57069856f
commit f737271303
1 changed files with 2 additions and 1 deletions

3
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