Update src/tools/plots.py

src/tools/plots.py

@@ -175,9 +175,10 @@ def create_lat_score_mem_plot_obj(leaderboard_df):
     copy_df["size"] = copy_df["size"].apply(lambda x: 0.8 if 0.8 <= x < 2 else x)
     copy_df["size"] = copy_df["size"].apply(lambda x: 1.5 if 2 <= x < 5 else x)
     copy_df["size"] = copy_df["size"].apply(lambda x: 2.0 if 5 <= x < 10 else x)
-    copy_df["size"] = copy_df["size"].apply(lambda x: 3.0 if 10 <= x < 20 else x)
-    copy_df["size"] = copy_df["size"].apply(lambda x: 4.5 if 20 <= x < 40 else x)
-    copy_df["size"] = copy_df["size"].apply(lambda x: 7.0 if x >= 40 else x)
+    copy_df["size"] = copy_df["size"].apply(lambda x: 3.0 if 10 <= x < 35 else x)
+    copy_df["size"] = copy_df["size"].apply(lambda x: 4.0 if 35 <= x < 60 else x)
+    copy_df["size"] = copy_df["size"].apply(lambda x: 6.0 if 60 <= x < 90 else x)
+    copy_df["size"] = copy_df["size"].apply(lambda x: 8.0 if x >= 90 else x)
 
     fig = px.scatter(
         copy_df,