Update src/saving_utils.py

src/saving_utils.py

@@ -15,8 +15,8 @@ def get_baseline_df(selected_methods, selected_metrics, leaderboard_path="/tmp/l
     leaderboard_df = pd.read_csv(leaderboard_path)
 
     if selected_methods is not None and selected_metrics is not None:
-        present_columns = ["method_name"] + selected_metrics
-        leaderboard_df = leaderboard_df[leaderboard_df['method_name'].isin(selected_methods)][present_columns]
+        present_columns = ["Method"] + selected_metrics
+        leaderboard_df = leaderboard_df[leaderboard_df['Method'].isin(selected_methods)][present_columns]
     return leaderboard_df