Introduce class attr BaseViz.enforce_numerical_metrics (#5176)

This commit is contained in:
Maxime Beauchemin 2018-06-13 22:45:03 -07:00 committed by GitHub
parent 30111bf3fd
commit a1095437bc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions

View File

@ -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:
@ -209,7 +210,8 @@ class BaseViz(object):
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)
self.handle_nulls(df)
@ -483,6 +485,7 @@ class TableViz(BaseViz):
verbose_name = _('Table View')
credits = 'a <a href="https://github.com/airbnb/superset">Superset</a> original'
is_timeseries = False
enforce_numerical_metrics = False
def should_be_timeseries(self):
fd = self.form_data