Update src/saving_utils.py

src/saving_utils.py

@@ -6,6 +6,19 @@ from huggingface_hub import HfApi
 script_dir = os.path.dirname(os.path.abspath(__file__))  # Directory of the running script
 
 
+
+def get_baseline_df(selected_methods, selected_metrics, leaderboard_path="/tmp/leaderboard_results.csv"):
+    if not os.path.exists(leaderboard_path):
+        benchmark_types = [] #only download leaderboard
+        download_from_hub(benchmark_types)
+
+    leaderboard_df = pd.read_csv(leaderboard_path)
+
+    present_columns = ["method_name"] + selected_metrics
+    leaderboard_df = leaderboard_df[leaderboard_df['method_name'].isin(selected_methods)][present_columns]
+    return leaderboard_df
+
+
 def save_results(method_name, benchmark_types, results, repo_id="mgyigit/probe-data", repo_type="space"):
     #First, download files to be updated from {repo_id}
     download_from_hub(benchmark_types, repo_id, repo_type)