update

app.py

@@ -116,21 +116,33 @@ def update_leaderboard_dataset_parallel(rl_env, path):
         return row
 
     data = list(thread_map(process_model, model_ids, desc="Processing models"))
-
     # Filter out None results (models with no metadata)
     data = [row for row in data if row is not None]
-
     ranked_dataframe = rank_dataframe(pd.DataFrame.from_records(data))
     new_history = ranked_dataframe
     file_path = path + "/" + rl_env + ".csv"
     new_history.to_csv(file_path, index=False)
-
     return ranked_dataframe
 
 def download_leaderboard_dataset():
     path = snapshot_download(repo_id=DATASET_REPO_ID, repo_type="dataset")
     return path
 
+path_ = download_leaderboard_dataset()
+
+def run_update_dataset():
+    global path_
+    for i in range(0, len(rl_envs)):
+        rl_env = rl_envs[i]
+        update_leaderboard_dataset_parallel(rl_env["rl_env"], path_)
+
+    api.upload_folder(
+    folder_path=path_,
+    repo_id="huph22/drlc-leaderboard-data",
+    repo_type="dataset",
+    commit_message="Update dataset")
+    
+
 def get_data(rl_env, path) -> pd.DataFrame:
     """
     Get data from rl_env
@@ -144,7 +156,6 @@ def get_data(rl_env, path) -> pd.DataFrame:
         data.loc[index, "User"] = make_clickable_user(user_id)
         model_id = row["Model"]
         data.loc[index, "Model"] = make_clickable_model(model_id)
-        
     return data
 
 def get_data_no_html(rl_env, path) -> pd.DataFrame:
@@ -154,7 +165,6 @@ def get_data_no_html(rl_env, path) -> pd.DataFrame:
     """
     csv_path = path + "/" + rl_env + ".csv"
     data = pd.read_csv(csv_path)
-
     return data
     
 def rank_dataframe(dataframe):
@@ -165,21 +175,6 @@ def rank_dataframe(dataframe):
         dataframe['Ranking'] =   [i for i in range(1,len(dataframe)+1)]
     return dataframe
 
-path_ = None
-
-def run_update_dataset():
-    global path_
-    path_ = download_leaderboard_dataset()
-    for i in range(0, len(rl_envs)):
-        rl_env = rl_envs[i]
-        update_leaderboard_dataset_parallel(rl_env["rl_env"], path_)
-
-    api.upload_folder(
-    folder_path=path_,
-    repo_id="huph22/drlc-leaderboard-data",
-    repo_type="dataset",
-    commit_message="Update dataset")
-
 def filter_data(rl_env, path, user_id):
     data_df = get_data_no_html(rl_env, path)
     models = []