update leader info fix

app.py

@@ -4,7 +4,7 @@ import pandas as pd
 import os
 import time
 import datetime
-from utils import fetch_competition_summary, fetch_models_evaluation, get_all_competition_summary, update_leader_info, highlight_score_column
+from utils import fetch_competition_summary, fetch_models_evaluation, get_all_competition_summary, highlight_score_column
 # from dotenv import load_dotenv
 from cfg import * 
 # load_dotenv()
@@ -28,7 +28,28 @@ else:
 
 wandb_api = wandb.Api()
 
-
+@st.cache_data
+def update_leader_info(leader_info, competition, best_model):
+ if leader_info.get(competition) is None:
+ leader_info[competition] = {
+ "Username": best_model["Username"],
+ "Hotkey": best_model["Hotkey"],
+ "Date": time.strftime("%Y-%m-%d"),
+ "Days on Top": 1
+ }
+ else:
+ if leader_info[competition]["Hotkey"] == best_model["Hotkey"]:
+ # count the number of days on top
+ start_date = datetime.datetime.strptime(leader_info[competition]["Date"], "%Y-%m-%d")
+ current_date = datetime.datetime.now()
+ days_on_top = (current_date - start_date).days
+ leader_info[competition]["Days on Top"] = days_on_top + 1
+ else:
+ leader_info[competition]["Username"] = best_model["Username"]
+ leader_info[competition]["Hotkey"] = best_model["Hotkey"]
+ leader_info[competition]["Date"] = time.strftime("%Y-%m-%d")
+ leader_info[competition]["Days on Top"] = 1
+ return leader_info[competition]
 
 @st.cache_data()
 def load_competition_data(last_update_time=None):

utils.py

@@ -1,31 +1,11 @@
 import streamlit as st
 import pandas as pd
 import wandb
+import time
 from datetime import datetime
 
 
 
-def update_leader_info(leader_info, competition, best_model):
- current_date = datetime.now().strftime("%Y-%m-%d")
- if leader_info.get(competition) is None:
- leader_info[competition] = {
- "Username": best_model["Username"],
- "Hotkey": best_model["Hotkey"],
- "Date": current_date,
- "Days on Top": 1
- }
- else:
- if leader_info[competition]["UID"] == best_model["ID"]:
- leader_info[competition]["Days on Top"] += 1
- else:
- leader_info[competition] = {
- "Username": best_model["Username"],
- "Hotkey": best_model["Hotkey"],
- "Date": current_date,
- "Days on Top": 1
- }
- return leader_info[competition]
-
 
 def get_all_competition_summary(api, projects):