diff --git a/superset/viz.py b/superset/viz.py index 6b5b68c38f514..cd1170984e325 100644 --- 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')