Skip to content

Commit cc0979e

Browse files
committed
Changig Handler's name
1 parent 692d85c commit cc0979e

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

causalpy/experiments/interrupted_time_series.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434
LEGEND_FONT_SIZE = 12
3535

3636

37-
class HandlerUTT:
37+
class UnknownTreatmentTimeHandler:
3838
"""
3939
A utility class for managing data preprocessing, postprocessing,
4040
and plotting steps for models that infer unknown treatment times.
@@ -185,7 +185,7 @@ def plot_intervention_line(
185185
)
186186

187187

188-
class HandlerKTT:
188+
class KnownTreatmentTimeHandler:
189189
"""
190190
Handles data preprocessing, postprocessing, and plotting logic for models
191191
where the treatment time is known in advance.
@@ -251,7 +251,7 @@ def plot_treated_counterfactual(
251251
self, sax, handles, labels, datapre, datapost, pre_pred, post_pred
252252
):
253253
"""
254-
Placeholder method to maintain interface compatibility with HandlerUTT.
254+
Placeholder method to maintain interface compatibility with UnknownTreatmentTimeHandler.
255255
"""
256256
pass
257257

@@ -342,9 +342,9 @@ def __init__(
342342

343343
# Getting the right handler
344344
if treatment_time is None or isinstance(treatment_time, tuple):
345-
self.handler = HandlerUTT()
345+
self.handler = UnknownTreatmentTimeHandler()
346346
else:
347-
self.handler = HandlerKTT()
347+
self.handler = KnownTreatmentTimeHandler()
348348

349349
# Preprocessing based on handler type
350350
self.datapre = self.handler.data_preprocessing(

0 commit comments

Comments
 (0)