Drop meaningless "level" columns from the correct dataframe.

(Does not affect performance) PiperOrigin-RevId: 313763435 Change-Id: I8973213e969be65a0a59ca73b07a1b8a628751a2
google-deepmind · May 29, 2020 · 876be3a · 876be3a
1 parent 933e386
commit 876be3a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bsuite/experiments/summary_analysis.py b/bsuite/experiments/summary_analysis.py
@@ -143,7 +143,7 @@ def bsuite_score(df: pd.DataFrame,
   # Groupby has a habit of adding meaningless columns to dataframe.
   for col in df.columns:
     if col in ['level_0', 'level_1', 'level_2']:
-      df.drop(col, axis=1, inplace=True)
+      score_df.drop(col, axis=1, inplace=True)
   return score_df