Fixed the usage of timedelta where months or years were used as the i…

…nterval.
kayak · Oct 18, 2018 · 0934c09 · 0934c09
1 parent bd9f274
commit 0934c09
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/fireant/slicer/queries/special_cases.py b/fireant/slicer/queries/special_cases.py
@@ -1,6 +1,5 @@
-from datetime import timedelta
-
 import pandas as pd
+from dateutil.relativedelta import relativedelta
 
 from fireant.slicer.dimensions import DatetimeDimension
 from fireant.slicer.filters import RangeFilter
@@ -50,7 +49,7 @@ def adjust_daterange_filter_for_rolling_window(dimensions, operations, filters):
         args = {dim0.interval + 's': max_rolling_period} \
             if 'quarter' != dim0.interval \
             else {'months': max_rolling_period * 3}
-        filter_.definition.start.value -= timedelta(**args)
+        filter_.definition.start.value -= relativedelta(**args)
 
     return filters
 

diff --git a/requirements.txt b/requirements.txt
@@ -3,5 +3,6 @@ pandas==0.22.0
 pypika==0.15.7
 toposort==1.5
 typing==3.6.2
+python-dateutil==2.7.3
 
 mock