bug #112: window size identification fixed

for trend change detection
zillow · Sep 15, 2022 · c6f1a63 · c6f1a63
1 parent 580318c
commit c6f1a63
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