From 876be3ab2f1dd1a294e559a8257fafbff3df4a97 Mon Sep 17 00:00:00 2001
From: Ian Osband <iosband@google.com>
Date: Fri, 29 May 2020 05:27:45 -0700
Subject: [PATCH] Drop meaningless "level" columns from the correct dataframe.
 (Does not affect performance)

PiperOrigin-RevId: 313763435
Change-Id: I8973213e969be65a0a59ca73b07a1b8a628751a2
---
 bsuite/experiments/summary_analysis.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bsuite/experiments/summary_analysis.py b/bsuite/experiments/summary_analysis.py
index b546cc4..e634db8 100644
--- 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