File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments