change return in recs function
This commit is contained in:
parent
150916879f
commit
f4e4ed02c7
4
main.py
4
main.py
@ -129,4 +129,6 @@ async def recs(production_id: str, count: int | None = 5):
|
||||
print(f'time elapsed = {time.time() - time_start}')
|
||||
scores = [idx[0] for idx in sorted(scores, key=lambda x: x[1], reverse=True)[:count+1]]
|
||||
scores.remove(production_id)
|
||||
return scores
|
||||
return {
|
||||
'id': scores
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user