Update

app.py

@@ -49,11 +49,13 @@ for issue in dataset["test"]:
 timeline_xs = []
 timeline_ys = []
 timeline_cols = []
+timeline_bugids = []
 model_cnt = 0
 for bug_id, time in bug_id_to_time.items():
     timeline_xs.append(time)
     timeline_ys.append(0)
     timeline_cols.append("All")
+    timeline_bugids.append(bug_id)
 LEADERBOARD_DF = get_leaderboard_df(EVAL_REQUESTS_PATH, total_issues)
 for row in LEADERBOARD_DF.itertuples():
     print(row)
@@ -62,11 +64,13 @@ for row in LEADERBOARD_DF.itertuples():
         timeline_xs.append(bug_id_to_time[fix])
         timeline_ys.append(-model_cnt)
         timeline_cols.append(row.method_id)
+        timeline_bugids.append(fix)
 timeline_df = pd.DataFrame(
     {
         "time": timeline_xs,
         "model": timeline_ys,
         "method_name": timeline_cols,
+        "bug_id": timeline_bugids,
     }
 )