Skip to content

Commit 942a1d5

Browse files
committed
fixing example
1 parent d00f828 commit 942a1d5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

causalpy/pymc_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -560,7 +560,7 @@ class InterventionTimeEstimator(PyMCModel):
560560
... coords={"obs_ind": data.index},
561561
... )
562562
>>> COORDS = {"coeffs":labels, "obs_ind": np.arange(_X.shape[0])}
563-
>>> model = ITE(sample_kwargs={"draws" : 10, "tune":10, "progressbar":False})
563+
>>> model = ITE(time_variable_name="t", sample_kwargs={"draws" : 10, "tune":10, "progressbar":False})
564564
>>> model.set_time_range(None, data)
565565
>>> model.fit(X=_X, y=_y, coords=COORDS)
566566
Inference ...

0 commit comments

Comments
 (0)