Merge pull request #591 from stan-dev/refactor-utilities

Split utils module into separate files

Author: WardBrian

cmdstanpy/__init__.py

@@ -6,18 +6,6 @@
 import shutil
 import tempfile
 
-_STANSUMMARY_STATS = [
-    'Mean',
-    'MCSE',
-    'StdDev',
-    '5%',
-    '50%',
-    '95%',
-    'N_Eff',
-    'N_Eff/s',
-    'R_hat',
-]
-
 _TMPDIR = tempfile.mkdtemp()
 _CMDSTAN_WARMUP = 1000
 _CMDSTAN_SAMPLING = 1000
@@ -38,20 +26,20 @@ def _cleanup_tmpdir() -> None:
 
 from ._version import __version__  # noqa
 from .install_cmdstan import rebuild_cmdstan
-from .model import CmdStanModel  # noqa
-from .stanfit import (  # noqa
+from .model import CmdStanModel
+from .stanfit import (
     CmdStanGQ,
     CmdStanMCMC,
     CmdStanMLE,
     CmdStanVB,
     InferenceMetadata,
     from_csv,
 )
-from .utils import set_cmdstan_path  # noqa
 from .utils import (
     cmdstan_path,
     cmdstan_version,
     install_cmdstan,
+    set_cmdstan_path,
     set_make_env,
     show_versions,
     write_stan_json,

cmdstanpy/cmdstan_args.py

@@ -830,7 +830,7 @@ def validate(self) -> None:
                     '0 and 2**32-1, found {}.'.format(self.seed)
                 )
             if isinstance(self.seed, int):
-                if self.seed < 0 or self.seed > 2**32 - 1:
+                if self.seed < 0 or self.seed > 2 ** 32 - 1:
                     raise ValueError(
                         'Argument "seed" must be an integer between '
                         '0 and 2**32-1, found {}.'.format(self.seed)
@@ -849,7 +849,7 @@ def validate(self) -> None:
                         )
                     )
                 for seed in self.seed:
-                    if seed < 0 or seed > 2**32 - 1:
+                    if seed < 0 or seed > 2 ** 32 - 1:
                         raise ValueError(
                             'Argument "seed" must be an integer value'
                             ' between 0 and 2**32-1,'

cmdstanpy/install_cxx_toolchain.py

@@ -27,7 +27,7 @@
 from cmdstanpy.utils import pushd, validate_dir, wrap_url_progress_hook
 
 EXTENSION = '.exe' if platform.system() == 'Windows' else ''
-IS_64BITS = sys.maxsize > 2**32
+IS_64BITS = sys.maxsize > 2 ** 32
 
 
 def usage() -> None:

cmdstanpy/stanfit/__init__.py

@@ -12,7 +12,8 @@
 )
 from cmdstanpy.utils import check_sampler_csv, get_logger, scan_config
 
-from .mcmc import CmdStanGQ, CmdStanMCMC
+from .gq import CmdStanGQ
+from .mcmc import CmdStanMCMC
 from .metadata import InferenceMetadata
 from .mle import CmdStanMLE
 from .runset import RunSet