Skip to content

Commit 08463b5

Browse files
committed
Changing variables name in preprocessing
1 parent 704066d commit 08463b5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

causalpy/experiments/interrupted_time_series.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,13 +73,13 @@ class UnknownTreatmentTimeHandler(TreatmentTimeHandler):
7373
and visualization.
7474
"""
7575

76-
def data_preprocessing(self, data, treatment_time, model):
76+
def data_preprocessing(self, data, time_range, model):
7777
"""
7878
Preprocesses the input data by constraining the model's
7979
treatment time inference window.
8080
"""
8181
# Restrict model's treatment time inference to given range
82-
model.set_time_range(treatment_time, data)
82+
model.set_time_range(time_range, data)
8383
return data
8484

8585
def data_postprocessing(

0 commit comments

Comments
 (0)