andrewrreed HF staff commited on
Commit
993d09a
1 Parent(s): 6fbf558

fix groupby deprecation warning

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -117,10 +117,8 @@ def build_plot(min_score, max_models_per_month, toggle_annotations, set_selector
117
  filtered_df = df[(df["rating"] >= min_score)]
118
 
119
  filtered_df = (
120
- filtered_df.groupby(["Month-Year", "License"])
121
- .apply(
122
- lambda x: x.nlargest(max_models_per_month, "rating"), include_groups=True
123
- )
124
  .reset_index(drop=True)
125
  )
126
 
 
117
  filtered_df = df[(df["rating"] >= min_score)]
118
 
119
  filtered_df = (
120
+ filtered_df.groupby(["Month-Year", "License"], group_keys=False)
121
+ .apply(lambda x: x.nlargest(max_models_per_month, "rating"))
 
 
122
  .reset_index(drop=True)
123
  )
124