fix

app.py

@@ -89,7 +89,6 @@ def get_benchmark_df(benchmark="Succeeded-1xA100-80GB"):
     ].fillna(1)
     # fix nan values
     merged_df[merged_df["generate.energy_consumption(tokens/kWh)"] == 1] = "N/A"
-
     # add optimizations
     merged_df["optimizations"] = merged_df["backend.bettertransformer"].apply(
         lambda x: "BetterTransformer" if x else "None"
@@ -102,7 +101,7 @@ def get_benchmark_df(benchmark="Succeeded-1xA100-80GB"):
     merged_df.sort_values(by=SORTING_COLUMN, ascending=SORTING_ASCENDING, inplace=True)
     # drop duplicates
     merged_df.drop_duplicates(subset=NO_DUPLICATES_COLUMNS, inplace=True)
-    merged_df = merged_df[[*ALL_COLUMNS_DATATYPES, "generate.latency(s)"]]
+    merged_df = merged_df[[*list(ALL_COLUMNS_MAPPING.keys()), "generate.latency(s)"]]
     return merged_df