From 9a3c3119d7201524be4c71076235e939bf05debe Mon Sep 17 00:00:00 2001 From: Ronan Laker Date: Fri, 24 Nov 2023 18:17:47 +0000 Subject: [PATCH] changed splitting behaviour for PrePostFit --- causalpy/pymc_experiments.py | 4 ++-- causalpy/skl_experiments.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/causalpy/pymc_experiments.py b/causalpy/pymc_experiments.py index 1c07e189..30073e3c 100644 --- 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 index 9db34736..4ec5db65 100644 --- 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