diff --git a/superset/viz.py b/superset/viz.py index a32456d77..dcd924ba8 100644 --- a/superset/viz.py +++ b/superset/viz.py @@ -145,6 +145,10 @@ class BaseViz(object): """ pass + def handle_nulls(self, df): + fillna = self.get_fillna_for_columns(df.columns) + df = df.fillna(fillna) + def get_fillna_for_col(self, col): """Returns the value for use as filler for a specific Column.type""" if col: @@ -208,8 +212,7 @@ class BaseViz(object): 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) - df = df.fillna(fillna) + self.handle_nulls(df) return df @staticmethod @@ -2041,6 +2044,9 @@ class BaseDeckGLViz(BaseViz): credits = 'deck.gl' spatial_control_keys = [] + def handle_nulls(self, df): + pass + def get_metrics(self): self.metric = self.form_data.get('size') return [self.metric] if self.metric else []