diff --git a/numpyro_sts/base.py b/numpyro_sts/base.py
index 14e0711..3e29957 100644
--- a/numpyro_sts/base.py
+++ b/numpyro_sts/base.py
@@ -274,11 +274,11 @@ def predict(self, n: int, value: jnp.ndarray) -> "LinearTimeseries":
         Creates a "prediction" instance of self.
 
         Args:
-            n:
-            value:
+            n: Number of future predictions.
+            value: New start value.
 
         Returns:
-
+            Returns new instance of :class:`LinearTimeseries`.
         """
 
         future_model = LinearTimeseries(