BenchmarkBot commited on
Commit
e89d633
β€’
1 Parent(s): 7c4e3f0

fix bench_df modifications

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -120,24 +120,26 @@ def get_benchmark_df(benchmark="1xA100-80GB"):
120
 
121
  def get_benchmark_table(bench_df):
122
  # add * to quantized models score
123
- bench_df["best_score"] = bench_df.apply(
 
 
124
  lambda x: f"{x['best_score']}**" if x["quantized"] else x["best_score"],
125
  axis=1,
126
  )
127
 
128
  # sort
129
- bench_df.sort_values(by=SORTING_COLUMN, ascending=True, inplace=True)
130
  # filter
131
- bench_df = bench_df[list(ALL_COLUMNS_MAPPING.keys())]
132
  # rename
133
- bench_df.rename(columns=ALL_COLUMNS_MAPPING, inplace=True)
134
  # transform
135
- bench_df["Type πŸ€—"] = bench_df["Type πŸ€—"].apply(process_model_type)
136
- bench_df["Best Scored Model πŸ†"] = bench_df["Best Scored Model πŸ†"].apply(
137
  process_model_name
138
  )
139
 
140
- return bench_df
141
 
142
 
143
  def get_benchmark_plot(bench_df):
 
120
 
121
  def get_benchmark_table(bench_df):
122
  # add * to quantized models score
123
+ copy_df = bench_df.copy()
124
+
125
+ copy_df["best_score"] = copy_df.apply(
126
  lambda x: f"{x['best_score']}**" if x["quantized"] else x["best_score"],
127
  axis=1,
128
  )
129
 
130
  # sort
131
+ copy_df.sort_values(by=SORTING_COLUMN, ascending=True, inplace=True)
132
  # filter
133
+ copy_df = copy_df[list(ALL_COLUMNS_MAPPING.keys())]
134
  # rename
135
+ copy_df.rename(columns=ALL_COLUMNS_MAPPING, inplace=True)
136
  # transform
137
+ copy_df["Type πŸ€—"] = copy_df["Type πŸ€—"].apply(process_model_type)
138
+ copy_df["Best Scored Model πŸ†"] = copy_df["Best Scored Model πŸ†"].apply(
139
  process_model_name
140
  )
141
 
142
+ return copy_df
143
 
144
 
145
  def get_benchmark_plot(bench_df):