Sort Leaderboard

play_with_hf.py

@@ -111,6 +111,11 @@ with (demo.route("Leaderboards", "/leaderboards") as demo_leaderboard):
                 'Time': [8888.8888],
             })
 
+        def sort_df(_cur_df):
+            return _cur_df.sort_values(["Solved", "Attempts", "Time"], key=lambda c: {
+                    "Solved": lambda s: -s.apply(len),
+                }.get(c.name, lambda s: s)(c))
+
         if not os.path.exists(_leaderboards):
             for lv in ['1', '2', '3']:
                 ret_leaderboards[lv] = add_dummies()
@@ -135,14 +140,11 @@ with (demo.route("Leaderboards", "/leaderboards") as demo_leaderboard):
                     })
                     return ret
                 flat = _cur_df.groupby("Session").apply(_per_session)
-                srt = flat.sort_values(["Solved", "Attempts", "Time"], key=lambda c: {
-                    "Solved": lambda s: -s.apply(len),
-                }.get(c.name, lambda s: s)(c))
-                return srt.iloc[0]
+                return sort_df(flat).iloc[0]
 
             for lv in ['1', '2', '3']:
                 cur_df = df_leaderboards_all.loc[df_leaderboards_all.Level.eq(lv)].groupby("User").apply(get_best)
-                ret_leaderboards[lv] = (cur_df.reset_index() if len(cur_df) else add_dummies()).rename({"Attempts": "Turns"}, axis=1)
+                ret_leaderboards[lv] = (sort_df(cur_df.reset_index()) if len(cur_df) else add_dummies()).rename({"Attempts": "Turns"}, axis=1)
 
         return ret_leaderboards