From f06c442e29d0888c8864c67dde51e972e31fba32 Mon Sep 17 00:00:00 2001
From: Victor <victor.tingstrom@gmail.com>
Date: Sat, 2 Nov 2024 10:14:54 +0100
Subject: [PATCH] Docs

---
 numpyro_sts/base.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

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(