From e41858ef15487e0ee7d3e77832b09b1627e533c7 Mon Sep 17 00:00:00 2001 From: Maxime Beauchemin Date: Wed, 13 Jun 2018 22:45:03 -0700 Subject: [PATCH] Introduce class attr BaseViz.enforce_numerical_metrics (#5176) (cherry picked from commit a1095437bc11fa29986f8d283a887d44343b22d3) --- superset/viz.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/superset/viz.py b/superset/viz.py index f61a7f4eb4b20..adb3efdc913ec 100644 --- a/superset/viz.py +++ b/superset/viz.py @@ -59,6 +59,7 @@ class BaseViz(object): is_timeseries = False default_fillna = 0 cache_type = 'df' + enforce_numerical_metrics = True def __init__(self, datasource, form_data, force=False): if not datasource: @@ -205,7 +206,8 @@ def get_df(self, query_obj=None): df[DTTM_ALIAS] += timedelta(hours=self.datasource.offset) df[DTTM_ALIAS] += self.time_shift - self.df_metrics_to_num(df, query_obj.get('metrics') or []) + if self.enforce_numerical_metrics: + self.df_metrics_to_num(df, query_obj.get('metrics') or []) df.replace([np.inf, -np.inf], np.nan) fillna = self.get_fillna_for_columns(df.columns) @@ -480,6 +482,7 @@ class TableViz(BaseViz): verbose_name = _('Table View') credits = 'a Superset original' is_timeseries = False + enforce_numerical_metrics = False def should_be_timeseries(self): fd = self.form_data