Fix lint

Author: ahartikainen

test/test_generate_quantities.py

@@ -83,7 +83,7 @@ def test_from_csv_files(self):
             bern_gqs.mcmc_sample.draws_pd().shape[1]
             + bern_gqs.draws_pd().shape[1],
         )
-        
+
         self.assertEqual(
             list(bern_gqs.draws_pd(vars=column_names).columns),
             column_names,
@@ -101,11 +101,12 @@ def test_from_csv_files(self):
             column_names[:3][::-1],
         )
         self.assertEqual(
-            list(bern_gqs.draws_pd(vars=['y_rep[3]', 'y_rep[6]', 'y_rep[2]']).columns),
+            list(bern_gqs.draws_pd(
+                vars=['y_rep[3]', 'y_rep[6]', 'y_rep[2]']
+            ).columns),
             ['y_rep[3]', 'y_rep[6]', 'y_rep[2]'],
         )
 
-
     def test_from_csv_files_bad(self):
         # gq model
         stan = os.path.join(DATAFILES_PATH, 'bernoulli_ppc.stan')