Skip to content

Commit 0645f28

Browse files
committed
fix user level for users without stats
1 parent 1141267 commit 0645f28

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

backend/oasst_backend/user_stats_repository.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,10 @@ def get_thresholds(baseline: int = 3, alpha: float = 1.1521, max_level: int = 10
5151
def _create_user_score(r, highlighted_user_id: UUID | None) -> UserScore:
5252
if r["UserStats"]:
5353
d = r["UserStats"].dict()
54+
d["level"] = (THRESHOLDS <= d["leader_score"]).sum()
5455
else:
5556
d = {"modified_date": utcnow()}
57+
d["level"] = 0
5658
for k in [
5759
"user_id",
5860
"username",
@@ -65,7 +67,7 @@ def _create_user_score(r, highlighted_user_id: UUID | None) -> UserScore:
6567
d[k] = r[k]
6668
if highlighted_user_id:
6769
d["highlighted"] = r["user_id"] == highlighted_user_id
68-
d["level"] = (THRESHOLDS <= d["leader_score"]).sum()
70+
6971
return UserScore(**d)
7072

7173

0 commit comments

Comments
 (0)