Minor fix in docstring

Author: JeanVanDyk

causalpy/pymc_models.py

@@ -541,7 +541,8 @@ class InterventionTimeEstimator(PyMCModel):
         ...     t,
         ...     y,
         ...     coords,
-        ...     effect=["impulse"])
+        ...     effect=["impulse"]
+        ...     n=5000)
         Inference data...
     """
 
@@ -600,11 +601,11 @@ def build_model(self, t, y, coords, effect, span, grain_season):
             # --- Likelihood ---
             pm.Normal("y_hat", mu=mu, sigma=2, observed=y)
 
-    def fit(self, t, y, coords, effect=[], span=None, grain_season=1):
+    def fit(self, t, y, coords, effect=[], span=None, grain_season=1, n=1000):
         """
         Draw samples from posterior distribution
         """
         self.build_model(t, y, coords, effect, span, grain_season)
         with self:
-            self.idata = pm.sample(**self.sample_kwargs)
+            self.idata = pm.sample(n, **self.sample_kwargs)
         return self.idata