add id column in tests

paris-saclay-cds · Sep 20, 2023 · 40bf0dd · 40bf0dd
1 parent d82792d
commit 40bf0dd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/ramp-database/ramp_database/tools/leaderboard.py b/ramp-database/ramp_database/tools/leaderboard.py
@@ -439,7 +439,7 @@ def get_leaderboard(
             columns = [
                 "team",
                 "submission",
-                "submission_id",
+                "id",
                 "submitted at (UTC)",
                 "state",
                 "waiting list",
@@ -448,7 +448,7 @@ def get_leaderboard(
             columns = [
                 "team",
                 "submission",
-                "submission_id",
+                "id",
                 "submitted at (UTC)",
                 "error",
             ]
@@ -487,7 +487,7 @@ def create_button(cell_value):
                 return f"<button onclick=\"alert(('You killed ' \
                     + 'submission {cell_value}!'))\">{cell_value}</button>"
 
-            df["Kill Sub"] = df["submission_id"].apply(
+            df["Stop"] = df["id"].apply(
                 lambda x: create_button(x)
             )
     else:

diff --git a/ramp-database/ramp_database/tools/tests/test_leaderboard.py b/ramp-database/ramp_database/tools/tests/test_leaderboard.py
@@ -257,6 +257,7 @@ def test_get_leaderboard(session_toy_db):
     assert (
         """<th>team</th>
       <th>submission</th>
+      <th>id</th>
       <th>submitted at (UTC)</th>
       <th>error</th>"""
         in leaderboard_failed