pt
/
superset
mirror of https://github.com/apache/superset.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix Pandas 0.24 DateOffset bug pt. 2 (#7981) 

* Fix pandas 0.24 DateOffset bug

* Add try-catch for DateOffsets that don't support normalize
This commit is contained in:
Ville Brofeldt 2019-08-08 00:17:23 +03:00 committed by Beto Dealmeida
parent cd6de3a1d8
commit 8cd8ec16d5
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
superset/viz.py
View File

@ -42,7 +42,6 @@ from markdown import markdown
import numpy as np
import pandas as pd
from pandas.tseries.frequencies import to_offset
from pandas.tseries.offsets import DateOffset
import polyline
import simplejson as json
@ -1402,7 +1401,10 @@ class NVD3TimePivotViz(NVD3TimeSeriesViz):
fd = self.form_data
df = self.process_data(df)
freq = to_offset(fd.get("freq"))
freq = DateOffset(normalize=True, **freq.kwds)
try:
freq = type(freq)(freq.n, normalize=True, **freq.kwds)
except ValueError:
freq = type(freq)(freq.n, **freq.kwds)
df.index.name = None
df[DTTM_ALIAS] = df.index.map(freq.rollback)
df["ranked"] = df[DTTM_ALIAS].rank(method="dense", ascending=False) - 1