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

This commit is contained in:
Maxime Beauchemin 2018-06-21 18:04:56 -07:00 committed by GitHub
parent 2f274328ed
commit 5a716e9925
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions

View File

@ -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')