From c6edcba0cf295a35d36af9944b22e90dc9342b6a Mon Sep 17 00:00:00 2001
From: xiaolongsun <95378566+xiaolong0728@users.noreply.github.com>
Date: Mon, 4 Nov 2024 14:13:34 +0100
Subject: [PATCH] Small fix to adapt to cm models

---
 common_utils/views_stepshifter_darts/hurdle_model.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/common_utils/views_stepshifter_darts/hurdle_model.py b/common_utils/views_stepshifter_darts/hurdle_model.py
index 986cc0b6..543e5281 100644
--- a/common_utils/views_stepshifter_darts/hurdle_model.py
+++ b/common_utils/views_stepshifter_darts/hurdle_model.py
@@ -43,6 +43,7 @@ def __init__(self, config: Dict, partitioner_dict: Dict[str, List[int]]):
 
     @views_validate
     def fit(self, df: pd.DataFrame, threshold: float = 0.1):
+        df = self._process_data(df)
         self._prepare_time_series(df)
 
         # Binary outcome (event/no-event)