Merge branch 'main' into round_to

Author: drbenvincent

causalpy/pymc_experiments.py

@@ -167,8 +167,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

@@ -76,8 +76,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