Updating doctest

Author: JeanVanDyk

causalpy/__init__.py

@@ -19,6 +19,7 @@
 from causalpy.version import __version__
 
 from .data import load_data
+from .experiments.change_point_detection import ChangePointDetection
 from .experiments.diff_in_diff import DifferenceInDifferences
 from .experiments.instrumental_variable import InstrumentalVariable
 from .experiments.interrupted_time_series import InterruptedTimeSeries
@@ -33,6 +34,7 @@
 __all__ = [
     "__version__",
     "DifferenceInDifferences",
+    "ChangePointDetection",
     "create_causalpy_compatible_class",
     "InstrumentalVariable",
     "InterruptedTimeSeries",

causalpy/experiments/change_point_detection.py

@@ -128,7 +128,7 @@ class ChangePointDetection(BaseExperiment):
     ...     .set_index("date")
     ... )
     >>> seed = 42
-    >>> result = cp.experiments.change_point_detection.ChangePointDetection(
+    >>> result = cp.ChangePointDetection(
     ...     df,
     ...     time_range=None,
     ...     formula="y ~ 1 + t + C(month)",