Update src/saving_utils.py

src/saving_utils.py

@@ -14,8 +14,9 @@ def get_baseline_df(selected_methods, selected_metrics, leaderboard_path="/tmp/l
 
     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]
+    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]
     return leaderboard_df