Updating integration test

JeanVanDyk · JeanVanDyk · commit 59db68976c1e · 2025-07-08T16:03:22.000+02:00
diff --git a/causalpy/tests/test_integration_pymc_examples.py b/causalpy/tests/test_integration_pymc_examples.py
@@ -427,7 +427,7 @@ def test_its_no_treatment_time():
         formula="standardize(deaths) ~ 0 + t + C(month) + standardize(temp)",  # noqa E501
         model=cp.pymc_models.InterventionTimeEstimator(
             time_variable_name="t",
-            treatment_type_effect={"impulse": [], "level": [], "trend": []},
+            treatment_effect_type=["impulse", "level", "trend"],
             sample_kwargs=sample_kwargs,
         ),
     )

Original file line number	Diff line number	Diff line change
`@@ -427,7 +427,7 @@ def test_its_no_treatment_time():`
`427`	`427`	`formula="standardize(deaths) ~ 0 + t + C(month) + standardize(temp)", # noqa E501`
`428`	`428`	`model=cp.pymc_models.InterventionTimeEstimator(`
`429`	`429`	`time_variable_name="t",`
`430`		`- treatment_type_effect={"impulse": [], "level": [], "trend": []},`
	`430`	`+ treatment_effect_type=["impulse", "level", "trend"],`
`431`	`431`	`sample_kwargs=sample_kwargs,`
`432`	`432`	`),`
`433`	`433`	`)`