Skip to content

Commit d48f0c3

Browse files
committed
Minor fix in docstring
1 parent d739b4a commit d48f0c3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

causalpy/pymc_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -600,11 +600,11 @@ def build_model(self, t, y, coords, effect, span, grain_season):
600600
# --- Likelihood ---
601601
pm.Normal("y_hat", mu=mu, sigma=2, observed=y)
602602

603-
def fit(self, t, y, coords, effect=[], span=None, grain_season=1, n=None):
603+
def fit(self, t, y, coords, effect=[], span=None, grain_season=1):
604604
"""
605605
Draw samples from posterior distribution
606606
"""
607607
self.build_model(t, y, coords, effect, span, grain_season)
608608
with self:
609-
self.idata = pm.sample(n, **self.sample_kwargs)
609+
self.idata = pm.sample(**self.sample_kwargs)
610610
return self.idata

0 commit comments

Comments
 (0)