Update test and timeseries more

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
-    ) -> TensorVariable:
+    ) -> at.TensorVariable:
 
         mu = at.as_tensor_variable(floatX(mu))
         sigma = at.as_tensor_variable(floatX(sigma))

pymc/tests/test_distributions_timeseries.py

@@ -88,6 +88,7 @@ def test_grw_inference(self):
 
 
 class TestGRWScipy(td.TestMatchesScipy):
+    # TODO: Test LogP for different inits in its own function
 
     # TODO: Find issue that says GRW wont take vector 
     def test_grw_logp(self):
@@ -99,13 +100,14 @@ def grw_logp(value, mu, sigma):
             stats.norm.logpdf(stationary_series, mu, sigma).sum(),
             return logp
 
-        # TODO: Make base class static static method
+        # TODO: Make base class a static method
         self.check_logp(
             pm.GaussianRandomWalk,
             td.Vector(td.R, 10),
             {"mu": td.R, "sigma": td.Rplus, "steps": td.Nat},
             grw_logp,
             decimal=select_by_precision(float64=6, float32=1),
+            n_samples=1,
         )