Change formatting and remove commented out lines

Author: ricardoV94

pymc/distributions/timeseries.py

@@ -25,6 +25,7 @@
 from pymc.distributions.continuous import Flat, Normal, get_tau_sigma
 from pymc.distributions.dist_math import check_parameters
 from pymc.distributions.shape_utils import to_tuple
+from pymc.util import check_dist_not_registered
 
 __all__ = [
     "AR1",
@@ -36,8 +37,6 @@
     "MvStudentTRandomWalk",
 ]
 
-from pymc.util import check_dist_not_registered
-
 
 class GaussianRandomWalkRV(RandomVariable):
     """

pymc/tests/test_distributions.py

@@ -2635,6 +2635,7 @@ def ref_logp(value, mu, sigma, steps):
             decimal=select_by_precision(float64=6, float32=1),
         )
 
+
 class TestBound:
     """Tests for pm.Bound distribution"""
 

pymc/tests/test_distributions_timeseries.py

@@ -14,8 +14,6 @@
 import numpy as np
 import pytest
 
-from scipy import stats
-
 import pymc as pm
 
 from pymc.aesaraf import floatX
@@ -26,11 +24,9 @@
     GARCH11,
     EulerMaruyama,
     GaussianRandomWalk,
-    gaussianrandomwalk,
 )
 from pymc.model import Model
 from pymc.sampling import sample, sample_posterior_predictive
-from pymc.tests import test_distributions as td
 from pymc.tests.helpers import select_by_precision
 from pymc.tests.test_distributions_random import BaseTestDistributionRandom
 
@@ -50,15 +46,8 @@ class TestGaussianRandomWalk(BaseTestDistributionRandom):
 
     def check_rv_inferred_size(self):
         steps = self.pymc_dist_params["steps"]
-        sizes_to_check = [
-            None,
-            (),
-            1,
-            (1,),
-        ]
+        sizes_to_check = [None, (), 1, (1,)]
         sizes_expected = [(steps + 1,), (steps + 1,), (1, steps + 1), (1, steps + 1)]
-        # sizes_to_check = [None, (), 1, (1,), 5, (4, 5), (2, 4, 2)]
-        # sizes_expected = [(), (), (1,), (1,), (5,), (4, 5), (2, 4, 2)]
 
         for size, expected in zip(sizes_to_check, sizes_expected):
             pymc_rv = self.pymc_dist.dist(**self.pymc_dist_params, size=size)
@@ -69,7 +58,7 @@ def check_not_implemented(self):
         with pytest.raises(NotImplementedError):
             self.pymc_rv.eval()
 
-    # TODO: Move this out of this test class, not related, makes more sense together with 
+    # TODO: Move this out of this test class, not related, makes more sense together with
     # the other logp tests
     def test_grw_inference(self):
         mu, sigma, steps = 2, 1, 10000