onekq commited on
Commit
501c623
1 Parent(s): 74c2db2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ with demo:
70
  filter_button.click(filter_data, inputs=[model_input, scenario_input], outputs=output)
71
 
72
  # Initialize leaderboard with the complete DataFrame
73
- complete_pass_at_k = df.groupby('Model', include_groups = False).apply(lambda x: pd.Series({
74
  'pass@1': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 1).mean(),
75
  'pass@5': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 5).mean(),
76
  'pass@10': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 10).mean()
 
70
  filter_button.click(filter_data, inputs=[model_input, scenario_input], outputs=output)
71
 
72
  # Initialize leaderboard with the complete DataFrame
73
+ complete_pass_at_k = df.groupby('Model')[['Runs', 'Successes']].apply(lambda x: pd.Series({
74
  'pass@1': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 1).mean(),
75
  'pass@5': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 5).mean(),
76
  'pass@10': estimate_pass_at_k(x['Runs'].values, x['Successes'].values, 10).mean()