BenchmarkBot commited on
Commit
06b3632
β€’
1 Parent(s): 6114ec7

fix drop duplicates

Browse files
Files changed (1) hide show
  1. app.py +11 -3
app.py CHANGED
@@ -58,6 +58,15 @@ ALL_COLUMNS_DATATYPES = [
58
  #
59
  "markdown",
60
  ]
 
 
 
 
 
 
 
 
 
61
  SORTING_COLUMN = ["best_score"]
62
  SORTING_ASCENDING = [False]
63
 
@@ -99,7 +108,8 @@ def get_benchmark_df(benchmark="Succeeded-1xA100-80GB"):
99
 
100
  # sort
101
  merged_df.sort_values(by=SORTING_COLUMN, ascending=SORTING_ASCENDING, inplace=True)
102
-
 
103
  return merged_df
104
 
105
 
@@ -122,8 +132,6 @@ def get_benchmark_table(bench_df):
122
  copy_df["Best Scored LLM πŸ†"] = copy_df["Best Scored LLM πŸ†"].apply(
123
  process_model_name
124
  )
125
- # drop duplicates
126
- copy_df.drop_duplicates(inplace=True)
127
 
128
  return copy_df
129
 
 
58
  #
59
  "markdown",
60
  ]
61
+ NO_DUPLICATES_COLUMNS = [
62
+ "backend.name",
63
+ "backend.torch_dtype",
64
+ "optimizations",
65
+ "quantization",
66
+ #
67
+ "weight_class",
68
+ "model_type",
69
+ ]
70
  SORTING_COLUMN = ["best_score"]
71
  SORTING_ASCENDING = [False]
72
 
 
108
 
109
  # sort
110
  merged_df.sort_values(by=SORTING_COLUMN, ascending=SORTING_ASCENDING, inplace=True)
111
+ # drop duplicates
112
+ merged_df.drop_duplicates(subset=NO_DUPLICATES_COLUMNS, inplace=True)
113
  return merged_df
114
 
115
 
 
132
  copy_df["Best Scored LLM πŸ†"] = copy_df["Best Scored LLM πŸ†"].apply(
133
  process_model_name
134
  )
 
 
135
 
136
  return copy_df
137