[bugfix] get word_cloud to support complex metrics (#5248)

apache · Jun 22, 2018 · 5a716e9 · 5a716e9
1 parent 2f27432
commit 5a716e9
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/superset/viz.py b/superset/viz.py
@@ -721,14 +721,13 @@ class WordCloudViz(BaseViz):
 
     def query_obj(self):
         d = super(WordCloudViz, self).query_obj()
-
-        d['metrics'] = [self.form_data.get('metric')]
         d['groupby'] = [self.form_data.get('series')]
         return d
 
     def get_data(self, df):
+        fd = self.form_data
         # Ordering the columns
-        df = df[[self.form_data.get('series'), self.form_data.get('metric')]]
+        df = df[[fd.get('series'), self.metric_labels[0]]]
         # Labeling the columns for uniform json schema
         df.columns = ['text', 'size']
         return df.to_dict(orient='records')