Merge pull request #275 from rlaker/prepost

changed splitting behaviour for PrePostFit

Author: drbenvincent

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
 

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