Merge pull request #1 from papaemman/bug/weekly-exploration-failure

bug zillow#112: window size identification fixed for trend change detection
papaemman · Sep 15, 2022 · c9008d5 · c9008d5
2 parents 580318c + c6f1a63
commit c9008d5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/luminaire/exploration/data_exploration.py b/luminaire/exploration/data_exploration.py
@@ -122,10 +122,10 @@ def __init__(self,
         tc_window_len_dict = {
             'H': 24,
             'D': 7,
+            'W':4
         }
 
-        if freq in ['H', 'D']:
-            self.tc_window_length = tc_window_len_dict.get(freq)
+        self.tc_window_length = tc_window_len_dict.get(freq) if freq in ['H', 'D', 'W'] else None
 
         self.tc_max_window_length = 24