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():
167
167
.set_index ("date" )
168
168
)
169
169
treatment_time = pd .to_datetime ("2020-01-01" )
170
- result = cp .pymc_experiments .SyntheticControl (
170
+ result = cp .pymc_experiments .InterruptedTimeSeries (
171
171
df ,
172
172
treatment_time ,
173
173
formula = "standardize(deaths) ~ 0 + standardize(t) + C(month) + standardize(temp)" , # noqa E501
174
174
model = cp .pymc_models .LinearRegression (sample_kwargs = sample_kwargs ),
175
175
)
176
176
assert isinstance (df , pd .DataFrame )
177
- assert isinstance (result , cp .pymc_experiments .SyntheticControl )
177
+ assert isinstance (result , cp .pymc_experiments .InterruptedTimeSeries )
178
178
assert len (result .idata .posterior .coords ["chain" ]) == sample_kwargs ["chains" ]
179
179
assert len (result .idata .posterior .coords ["draw" ]) == sample_kwargs ["draws" ]
180
180
You can’t perform that action at this time.
0 commit comments