Updating doctest

JeanVanDyk · JeanVanDyk · commit cdfd8e128ea0 · 2025-10-22T13:45:58.000+02:00
diff --git a/causalpy/tests/test_integration_pymc_examples.py b/causalpy/tests/test_integration_pymc_examples.py
@@ -438,7 +438,7 @@ def test_cp_covid():
         time_range=time_range,
         formula="standardize(deaths) ~ 0 + t + C(month) + standardize(temp)",  # noqa E501
         model=cp.pymc_models.LinearChangePointDetection(
-            treatment_effect_type=["impulse", "level", "trend"],
+            cp_effect_type=["impulse", "level", "trend"],
             sample_kwargs=sample_kwargs,
         ),
     )

Original file line number	Diff line number	Diff line change
`@@ -438,7 +438,7 @@ def test_cp_covid():`
`438`	`438`	`time_range=time_range,`
`439`	`439`	`formula="standardize(deaths) ~ 0 + t + C(month) + standardize(temp)", # noqa E501`
`440`	`440`	`model=cp.pymc_models.LinearChangePointDetection(`
`441`		`- treatment_effect_type=["impulse", "level", "trend"],`
	`441`	`+ cp_effect_type=["impulse", "level", "trend"],`
`442`	`442`	`sample_kwargs=sample_kwargs,`
`443`	`443`	`),`
`444`	`444`	`)`