Test set submission working

app.py

@@ -30,7 +30,6 @@ def get_leaderboard_object(assay: str | None = None):
     # TODO how to sort filter columns alphabetically?
     # Bug: Can't leave search_columns empty because then it says "Column None not found in headers"
     # Note(Lood): Would be nice to make it clear that the Search Column is searching on model name
-    # TODO(Lood) check that this actually refreshes using the function
     lb = Leaderboard(
         value=format_leaderboard_table(df_results=current_dataframe, assay=assay),
         datatype=["str", "str", "str", "number"],
@@ -161,7 +160,7 @@ with gr.Blocks() as demo:
                     )
                 with gr.Column():
                     submission_type_dropdown = gr.Dropdown(
-                        choices=["GDPa1", "GDPa1_cross_validation"],
+                        choices=["GDPa1", "GDPa1_cross_validation", "Heldout Test Set"],
                         value="GDPa1",
                         label="Submission Type",
                     )
@@ -179,7 +178,7 @@ with gr.Blocks() as demo:
                     Dynamically update example file for download
                 """
                 download_file = EXAMPLE_FILE_DICT.get(
-                    submission_type, EXAMPLE_FILE_DICT["GDPa1"]
+                    submission_type, EXAMPLE_FILE_DICT[submission_type]
                 )
                 download_label = (
                     f"📥 Download example submission CSV for {submission_type}"

constants.py

@@ -43,14 +43,14 @@ CV_COLUMN = "hierarchical_cluster_IgG_isotype_stratified_fold"
 EXAMPLE_FILE_DICT = {
     "GDPa1": "data/example-predictions.csv",
     "GDPa1_cross_validation": "data/example-predictions-cv.csv",
-    "heldout_test": "data/example-predictions-heldout.csv",
+    "Heldout Test Set": "data/example-predictions-heldout.csv",
 }
 ANTIBODY_NAMES_DICT = {
     "GDPa1": pd.read_csv(EXAMPLE_FILE_DICT["GDPa1"])["antibody_name"].tolist(),
     "GDPa1_cross_validation": pd.read_csv(EXAMPLE_FILE_DICT["GDPa1_cross_validation"])[
         "antibody_name"
     ].tolist(),
-    "heldout_test": pd.read_csv(EXAMPLE_FILE_DICT["heldout_test"])["antibody_name"].tolist(),   # TODO add a test for this validation
+    "Heldout Test Set": pd.read_csv(EXAMPLE_FILE_DICT["Heldout Test Set"])["antibody_name"].tolist(),
 }
 
 # Huggingface API