eduagarcia commited on
Commit
0cb9327
1 Parent(s): b74e881

fix eval time plot

Browse files
Files changed (1) hide show
  1. src/tools/plots.py +1 -1
src/tools/plots.py CHANGED
@@ -173,7 +173,7 @@ def create_lat_score_mem_plot_obj(leaderboard_df):
173
  copy_df["size"] = copy_df["size"].apply(lambda x: 2.0 if 5 <= x < 10 else x)
174
  copy_df["size"] = copy_df["size"].apply(lambda x: 3.0 if 10 <= x < 20 else x)
175
  copy_df["size"] = copy_df["size"].apply(lambda x: 4.5 if 20 <= x < 40 else x)
176
- copy_df["size"] = copy_df["size"].apply(lambda x: 7.0 if x > 40 else x)
177
 
178
  fig = px.scatter(
179
  copy_df,
 
173
  copy_df["size"] = copy_df["size"].apply(lambda x: 2.0 if 5 <= x < 10 else x)
174
  copy_df["size"] = copy_df["size"].apply(lambda x: 3.0 if 10 <= x < 20 else x)
175
  copy_df["size"] = copy_df["size"].apply(lambda x: 4.5 if 20 <= x < 40 else x)
176
+ copy_df["size"] = copy_df["size"].apply(lambda x: 7.0 if x >= 40 else x)
177
 
178
  fig = px.scatter(
179
  copy_df,