Skip to content

Commit 05d570d

Browse files
committed
update pymc integration test code
1 parent a06b99c commit 05d570d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

causalpy/tests/test_integration_pymc_examples.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -167,14 +167,14 @@ def test_its_covid():
167167
.set_index("date")
168168
)
169169
treatment_time = pd.to_datetime("2020-01-01")
170-
result = cp.pymc_experiments.SyntheticControl(
170+
result = cp.pymc_experiments.InterruptedTimeSeries(
171171
df,
172172
treatment_time,
173173
formula="standardize(deaths) ~ 0 + standardize(t) + C(month) + standardize(temp)", # noqa E501
174174
model=cp.pymc_models.LinearRegression(sample_kwargs=sample_kwargs),
175175
)
176176
assert isinstance(df, pd.DataFrame)
177-
assert isinstance(result, cp.pymc_experiments.SyntheticControl)
177+
assert isinstance(result, cp.pymc_experiments.InterruptedTimeSeries)
178178
assert len(result.idata.posterior.coords["chain"]) == sample_kwargs["chains"]
179179
assert len(result.idata.posterior.coords["draw"]) == sample_kwargs["draws"]
180180

0 commit comments

Comments
 (0)