Skip to content

Commit d739b4a

Browse files
committed
Minor fix in docstring
1 parent 3dc23b3 commit d739b4a

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

causalpy/pymc_models.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -541,8 +541,7 @@ class InterventionTimeEstimator(PyMCModel):
541541
... t,
542542
... y,
543543
... coords,
544-
... effect=["impulse"],
545-
... n=1000)
544+
... effect=["impulse"])
546545
Inference data...
547546
"""
548547

@@ -601,7 +600,7 @@ def build_model(self, t, y, coords, effect, span, grain_season):
601600
# --- Likelihood ---
602601
pm.Normal("y_hat", mu=mu, sigma=2, observed=y)
603602

604-
def fit(self, t, y, coords, effect=[], span=None, grain_season=1, n=1000):
603+
def fit(self, t, y, coords, effect=[], span=None, grain_season=1, n=None):
605604
"""
606605
Draw samples from posterior distribution
607606
"""

0 commit comments

Comments
 (0)