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

bug #112: support for weekly frequency
zillow · Sep 29, 2022 · 0af6a34 · 0af6a34
2 parents 580318c + c6f1a63
commit 0af6a34
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