andrewrreed HF staff commited on
Commit
6601027
1 Parent(s): 03480fc

suppress deprecation warning

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -115,7 +115,9 @@ def build_plot(min_score, max_models_per_month, toggle_annotations, set_selector
115
 
116
  filtered_df = (
117
  filtered_df.groupby(["Month-Year", "License"])
118
- .apply(lambda x: x.nlargest(max_models_per_month, "rating"))
 
 
119
  .reset_index(drop=True)
120
  )
121
 
 
115
 
116
  filtered_df = (
117
  filtered_df.groupby(["Month-Year", "License"])
118
+ .apply(
119
+ lambda x: x.nlargest(max_models_per_month, "rating"), include_groups=True
120
+ )
121
  .reset_index(drop=True)
122
  )
123