Merge pull request #275 from rlaker/prepost

changed splitting behaviour for PrePostFit
pymc-labs · Dec 22, 2023 · fc28a3b · fc28a3b
2 parents c17fcf2 + 0c3f23d
commit fc28a3b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/causalpy/pymc_experiments.py b/causalpy/pymc_experiments.py
@@ -164,8 +164,8 @@ def __init__(
         # set experiment type - usually done in subclasses
         self.expt_type = "Pre-Post Fit"
         # split data in to pre and post intervention
-        self.datapre = data[data.index <= self.treatment_time]
-        self.datapost = data[data.index > self.treatment_time]
+        self.datapre = data[data.index < self.treatment_time]
+        self.datapost = data[data.index >= self.treatment_time]
 
         self.formula = formula
 

diff --git a/causalpy/skl_experiments.py b/causalpy/skl_experiments.py
@@ -74,8 +74,8 @@ def __init__(
         super().__init__(model=model, **kwargs)
         self.treatment_time = treatment_time
         # split data in to pre and post intervention
-        self.datapre = data[data.index <= self.treatment_time]
-        self.datapost = data[data.index > self.treatment_time]
+        self.datapre = data[data.index < self.treatment_time]
+        self.datapost = data[data.index >= self.treatment_time]
 
         self.formula = formula