lint fix

Author: mitzimorris

cmdstanpy/model.py

@@ -1320,6 +1320,7 @@ def _run_cmdstan(
                 fd_out.write(stdout)
                 if show_console:
                     lines = stdout.split('\n')
+                    print(f' ****** lines to process: {len(lines)}')
                     for line in lines:
                         print('chain {}: {}'.format(idx + 1, stdout))
 

test/test_sample.py

@@ -332,8 +332,10 @@ def test_multi_proc(self):
         with LogCapture() as log:
             logging.getLogger()
             logistic_model.sample(
-                data=logistic_data, chains=4, parallel_chains=1,
-                show_progress=False
+                data=logistic_data,
+                chains=4,
+                parallel_chains=1,
+                show_progress=False,
             )
         log.check_present(
             ('cmdstanpy', 'INFO', 'finish chain 1'),
@@ -342,8 +344,10 @@ def test_multi_proc(self):
         with LogCapture() as log:
             logging.getLogger()
             logistic_model.sample(
-                data=logistic_data, chains=4, parallel_chains=2,
-                show_progress=False
+                data=logistic_data,
+                chains=4,
+                parallel_chains=2,
+                show_progress=False,
             )
         if cpu_count() >= 4:
             # finish chains 1, 2 before starting chains 3, 4
@@ -355,8 +359,10 @@ def test_multi_proc(self):
             with LogCapture() as log:
                 logging.getLogger()
                 logistic_model.sample(
-                    data=logistic_data, chains=4, parallel_chains=4,
-                    show_progress=False
+                    data=logistic_data,
+                    chains=4,
+                    parallel_chains=4,
+                    show_progress=False,
                 )
                 log.check_present(
                     ('cmdstanpy', 'INFO', 'start chain 4'),
@@ -370,7 +376,7 @@ def test_multi_proc(self):
                 chains=1,
                 parallel_chains=1,
                 threads_per_chain=7,
-                show_progress=False
+                show_progress=False,
             )
         log.check_present(('cmdstanpy', 'DEBUG', 'total threads: 7'))
         with LogCapture() as log:
@@ -380,7 +386,7 @@ def test_multi_proc(self):
                 chains=7,
                 parallel_chains=1,
                 threads_per_chain=5,
-                show_progress=False
+                show_progress=False,
             )
         log.check_present(('cmdstanpy', 'DEBUG', 'total threads: 5'))
         with LogCapture() as log:
@@ -581,13 +587,10 @@ def test_show_progress(self, stanfile='bernoulli.stan'):
         bern_model = CmdStanModel(stan_file=stan)
         jdata = os.path.join(DATAFILES_PATH, 'bernoulli.data.json')
 
-        sys_stderr = io.StringIO()   # tqdm prints to stderr
+        sys_stderr = io.StringIO()  # tqdm prints to stderr
         with contextlib.redirect_stderr(sys_stderr):
             bern_model.sample(
-                data=jdata,
-                chains=2,
-                parallel_chains=2,
-                show_progress=True
+                data=jdata, chains=2, parallel_chains=2, show_progress=True
             )
             console = sys_stderr.getvalue()
         self.assertTrue('chain 1' in console)

test/test_utils.py

@@ -765,7 +765,6 @@ def test_parse_containers(self):
 
 
 class DoCommandTest(unittest.TestCase):
-
     def test_capture_console(self):
         tmp = io.StringIO()
         do_command(cmd=['ls'], cwd=HERE, fd_out=tmp)
@@ -799,6 +798,7 @@ def test_bad(self):
         with self.assertRaisesRegex(ValueError, 'Expecting 3D array'):
             flatten_chains(array_2d)
 
+
 @pytest.mark.order(-1)
 class ShowProgressTest(unittest.TestCase):
     # this test must run after any tests that check tqdm progress bars
@@ -816,7 +816,7 @@ def test_show_progress_fns(self):
                 'ERROR',
                 'Error in progress bar initialization:\n'
                 '\terror\n'
-                'Disabling progress bars for this session'
+                'Disabling progress bars for this session',
             )
         )
         self.assertFalse(allow_show_progress())
@@ -834,8 +834,5 @@ def test_show_progress_fns(self):
         self.assertFalse(allow_show_progress())
 
 
-
-
-
 if __name__ == '__main__':
     unittest.main()