pymc-labs
diff --git a/‎causalpy/tests/test_integration_pymc_examples.py‎
Lines changed: 1 addition & 1 deletion b/‎causalpy/tests/test_integration_pymc_examples.py‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎docs/source/_static/classes.png‎
6.68 KB b/‎docs/source/_static/classes.png‎
6.68 KB
diff --git a/‎docs/source/notebooks/its_no_treatment_time.ipynb‎
Lines changed: 165 additions & 103 deletions b/‎docs/source/notebooks/its_no_treatment_time.ipynb‎
Lines changed: 165 additions & 103 deletions
@@ -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": []},
+            treatment_type_effect={"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": []},`
	`430`	`+ treatment_type_effect={"impulse": [], "level": [], "trend": []},`
`431`	`431`	`sample_kwargs=sample_kwargs,`
`432`	`432`	`),`
`433`	`433`	`)`