lint fix

Author: mitzimorris

cmdstanpy/stanfit/mcmc.py

@@ -77,13 +77,13 @@ def __init__(
         assert isinstance(
             sampler_args, SamplerArgs
         )  # make the typechecker happy
-        self._iter_sampling:int = _CMDSTAN_SAMPLING
+        self._iter_sampling: int = _CMDSTAN_SAMPLING
         if sampler_args.iter_sampling is not None:
             self._iter_sampling = sampler_args.iter_sampling
-        self._iter_warmup:int = _CMDSTAN_WARMUP
+        self._iter_warmup: int = _CMDSTAN_WARMUP
         if sampler_args.iter_warmup is not None:
             self._iter_warmup = sampler_args.iter_warmup
-        self._thin:int = _CMDSTAN_THIN
+        self._thin: int = _CMDSTAN_THIN
         if sampler_args.thin is not None:
             self._thin = sampler_args.thin
         self._is_fixed_param = sampler_args.fixed_param
@@ -286,8 +286,12 @@ def _validate_csv_files(self) -> Dict[str, Any]:
         Raises exception when inconsistencies detected.
         """
         if not self._is_fixed_param:
-            self._divergences: np.ndarray = np.zeros(self.runset.chains, dtype=int)
-            self._max_treedepths: np.ndarray = np.zeros(self.runset.chains, dtype=int)
+            self._divergences: np.ndarray = np.zeros(
+                self.runset.chains, dtype=int
+            )
+            self._max_treedepths: np.ndarray = np.zeros(
+                self.runset.chains, dtype=int
+            )
 
         dzero = {}
         for i in range(self.chains):
@@ -369,7 +373,7 @@ def _check_sampler_diagnostics(self) -> None:
                 'Use function "diagnose()" to see further information.'
             )
             get_logger().warning('\n\t'.join(diagnostics))
-    
+
     def _assemble_draws(self) -> None:
         """
         Allocates and populates the step size, metric, and sample arrays

test/test_sample.py

@@ -1808,11 +1808,15 @@ def test_diagnostics(self):
                 data=rdata,
                 seed=55157,
             )
-            log.check_present((
-                'cmdstanpy',
-                'WARNING',
-                StringComparison(r'(?s).*Some chains may have failed to converge.*')
-            ))
+            log.check_present(
+                (
+                    'cmdstanpy',
+                    'WARNING',
+                    StringComparison(
+                        r'(?s).*Some chains may have failed to converge.*'
+                    ),
+                )
+            )
             self.assertFalse(np.all(fit.divergences == 0))
 
         with LogCapture(level=logging.WARNING) as log:
@@ -1822,11 +1826,13 @@ def test_diagnostics(self):
                 seed=40508,
                 max_treedepth=3,
             )
-            log.check_present((
-                'cmdstanpy',
-                'WARNING',
-                StringComparison(r'(?s).*max treedepth*')
-            ))
+            log.check_present(
+                (
+                    'cmdstanpy',
+                    'WARNING',
+                    StringComparison(r'(?s).*max treedepth*'),
+                )
+            )
             self.assertFalse(np.all(fit.max_treedepths == 0))
 
         stan = os.path.join(DATAFILES_PATH, 'bernoulli.stan')