Skip to content

Commit 3eb24a9

Browse files
committed
use new API in scikit-learn integration test
1 parent 3d29fef commit 3eb24a9

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

causalpy/tests/test_integration_skl_examples.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,8 @@ def test_sc():
142142
result = cp.SyntheticControl(
143143
df,
144144
treatment_time,
145-
formula="actual ~ 0 + a + b + c + d + e + f + g",
145+
control_units=["a", "b", "c", "d", "e", "f", "g"],
146+
treated_units=["actual"],
146147
model=cp.skl_models.WeightedProportion(),
147148
)
148149
assert isinstance(df, pd.DataFrame)

0 commit comments

Comments
 (0)