apply black via pre-commit

Author: michaelosthege

pymc3/tests/test_memo.py

@@ -52,7 +52,10 @@ def test_hashing_of_rv_tuples():
         mu = pm.Normal("mu", 0, 1)
         sd = pm.Gamma("sd", 1, 2)
         dd = pm.DensityDist(
-            "dd", pm.Normal.dist(mu, sd).logp, random=pm.Normal.dist(mu, sd).random, observed=obs,
+            "dd",
+            pm.Normal.dist(mu, sd).logp,
+            random=pm.Normal.dist(mu, sd).random,
+            observed=obs,
         )
         print()
         for freerv in [mu, sd, dd] + pmodel.free_RVs:

pymc3/tests/test_sampling.py

@@ -72,14 +72,22 @@ def test_sample(self):
             for cores in test_cores:
                 for steps in [1, 10, 300]:
                     pm.sample(
-                        steps, tune=0, step=self.step, cores=cores, random_seed=self.random_seed,
+                        steps,
+                        tune=0,
+                        step=self.step,
+                        cores=cores,
+                        random_seed=self.random_seed,
                     )
 
     def test_sample_init(self):
         with self.model:
             for init in ("advi", "advi_map", "map"):
                 pm.sample(
-                    init=init, tune=0, n_init=1000, draws=50, random_seed=self.random_seed,
+                    init=init,
+                    tune=0,
+                    n_init=1000,
+                    draws=50,
+                    random_seed=self.random_seed,
                 )
 
     def test_sample_args(self):
@@ -99,7 +107,11 @@ def test_sample_args(self):
     def test_iter_sample(self):
         with self.model:
             samps = pm.sampling.iter_sample(
-                draws=5, step=self.step, start=self.start, tune=0, random_seed=self.random_seed,
+                draws=5,
+                step=self.step,
+                start=self.start,
+                tune=0,
+                random_seed=self.random_seed,
             )
             for i, trace in enumerate(samps):
                 assert i == len(trace) - 1, "Trace does not have correct length."

pymc3/util.py

@@ -289,7 +289,8 @@ def chains_and_samples(data: Union[xarray.Dataset, arviz.InferenceData]) -> Tupl
         dataset = data.posterior
     else:
         raise ValueError(
-            "Argument must be xarray Dataset or arviz InferenceData. Got %s", data.__class__,
+            "Argument must be xarray Dataset or arviz InferenceData. Got %s",
+            data.__class__,
         )
 
     coords = dataset.coords