Update pymc/distributions/timeseries.py

Co-authored-by: Ricardo Vieira <[email protected]>

Author: canyon289

pymc/distributions/timeseries.py

@@ -150,7 +150,7 @@ def __new__(cls, name, mu=0.0, sigma=1.0, init=None, steps: int = 1, **kwargs):
     @classmethod
     def dist(
         cls, mu=0.0, sigma=1.0, init=None, steps: int = 1, size=None, **kwargs
-    ) -> RandomVariable:
+    ) -> TensorVariable:
 
         mu = at.as_tensor_variable(floatX(mu))
         sigma = at.as_tensor_variable(floatX(sigma))