Reformat with black.

Author: tillahoffmann

cmdstanpy/model.py

@@ -727,8 +727,12 @@ def optimize(
             )
             dummy_chain_id = 0
             runset = RunSet(args=args, chains=1, time_fmt=time_fmt)
-            self._run_cmdstan(runset, dummy_chain_id, show_console=show_console,
-                              timeout=timeout)
+            self._run_cmdstan(
+                runset,
+                dummy_chain_id,
+                show_console=show_console,
+                timeout=timeout,
+            )
 
         if not runset._check_retcodes():
             msg = 'Error during optimization: {}'.format(runset.get_err_msgs())
@@ -1508,8 +1512,12 @@ def variational(
 
             dummy_chain_id = 0
             runset = RunSet(args=args, chains=1, time_fmt=time_fmt)
-            self._run_cmdstan(runset, dummy_chain_id, show_console=show_console,
-                              timeout=timeout)
+            self._run_cmdstan(
+                runset,
+                dummy_chain_id,
+                show_console=show_console,
+                timeout=timeout,
+            )
 
         # treat failure to converge as failure
         transcript_file = runset.stdout_files[dummy_chain_id]
@@ -1546,7 +1554,8 @@ def variational(
                 )
             else:
                 msg = 'Error during variational inference: {}'.format(
-                    runset.get_err_msgs())
+                    runset.get_err_msgs()
+                )
             raise RuntimeError(msg)
         # pylint: disable=invalid-name
         vb = CmdStanVB(runset)

test/test_generate_quantities.py

@@ -481,8 +481,9 @@ def test_timeout(self):
         timeout_model = CmdStanModel(stan_file=stan)
         fit = timeout_model.sample(data={'loop': 0}, chains=1, iter_sampling=10)
         with self.assertRaisesRegex(RuntimeError, 'processing timed out'):
-            timeout_model.generate_quantities(timeout=0.1, mcmc_sample=fit,
-                                              data={'loop': 1})
+            timeout_model.generate_quantities(
+                timeout=0.1, mcmc_sample=fit, data={'loop': 1}
+            )
 
 
 if __name__ == '__main__':