Skip to content

Commit 182aac0

Browse files
committed
get deprecation tests working again
1 parent 1f4d17e commit 182aac0

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

causalpy/tests/test_integration_pymc_examples.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -803,7 +803,8 @@ def test_sc_deprecation_warning():
803803
result = cp.pymc_experiments.SyntheticControl(
804804
df,
805805
treatment_time,
806-
formula="actual ~ 0 + a + b + c + d + e + f + g",
806+
control_units=["a", "b", "c", "d", "e", "f", "g"],
807+
treated_units=["actual"],
807808
model=cp.pymc_models.WeightedSumFitter(sample_kwargs=sample_kwargs),
808809
)
809810
assert isinstance(result, cp.SyntheticControl)

causalpy/tests/test_integration_skl_examples.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -324,7 +324,8 @@ def test_sc_deprecation_warning():
324324
result = cp.skl_experiments.SyntheticControl(
325325
df,
326326
treatment_time,
327-
formula="actual ~ 0 + a + b + c + d + e + f + g",
327+
control_units=["a", "b", "c", "d", "e", "f", "g"],
328+
treated_units=["actual"],
328329
model=cp.skl_models.WeightedProportion(),
329330
)
330331
assert isinstance(result, cp.SyntheticControl)

0 commit comments

Comments
 (0)