viz: make sunburst work again (#728)

By using a different method for renaming the metric columns Thanks to @simobasso for the help! Fix #673
apache · Jul 13, 2016 · 212284c · 212284c
1 parent 082645d
commit 212284c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/caravel/viz.py b/caravel/viz.py
@@ -1304,9 +1304,8 @@ def get_data(self):
         metric = self.form_data.get('metric')
         secondary_metric = self.form_data.get('secondary_metric')
         if metric == secondary_metric:
-            ndf = df[cols]
-            ndf['m1'] = df[metric]
-            ndf['m2'] = df[metric]
+            ndf = df
+            ndf.columns = [cols + ['m1', 'm2']]
         else:
             cols += [
                 self.form_data['metric'], self.form_data['secondary_metric']]