cmdstanpy 1.2.3__tar.gz → 1.2.4__tar.gz

{cmdstanpy-1.2.3/cmdstanpy.egg-info → cmdstanpy-1.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cmdstanpy
-Version: 1.2.3
+Version: 1.2.4
 Summary: Python interface to CmdStan
 Author: Stan Dev Team
 License: BSD-3-Clause

cmdstanpy-1.2.4/cmdstanpy/_version.py

@@ -0,0 +1,3 @@
+"""PyPi Version"""
+
+__version__ = '1.2.4'

{cmdstanpy-1.2.3 → cmdstanpy-1.2.4}/cmdstanpy/utils/stancsv.py

@@ -193,7 +193,12 @@ def scan_config(fd: TextIO, config_dict: Dict[str, Any], lineno: int) -> int:
                     try:
                         val = float(raw_val)
                     except ValueError:
-                        val = raw_val
+                        if raw_val == "true":
+                            val = 1
+                        elif raw_val == "false":
+                            val = 0
+                        else:
+                            val = raw_val
                 config_dict[key_val[0].strip()] = val
         cur_pos = fd.tell()
         line = fd.readline().strip()

{cmdstanpy-1.2.3 → cmdstanpy-1.2.4/cmdstanpy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cmdstanpy
-Version: 1.2.3
+Version: 1.2.4
 Summary: Python interface to CmdStan
 Author: Stan Dev Team
 License: BSD-3-Clause

{cmdstanpy-1.2.3 → cmdstanpy-1.2.4}/test/test_sample.py

@@ -2030,6 +2030,26 @@ def test_tuple_data_in() -> None:
     data_model.sample(data, chains=1, iter_warmup=1, iter_sampling=1)
 
 
+def test_csv_roundtrip():
+    stan = os.path.join(DATAFILES_PATH, 'matrix_var.stan')
+    model = CmdStanModel(stan_file=stan)
+    fit = model.sample(
+        iter_sampling=10, iter_warmup=9, chains=2, save_warmup=True
+    )
+    z = fit.stan_variable(var="z")
+    assert z.shape == (20, 4, 3)
+    z_with_warmup = fit.stan_variable(var="z", inc_warmup=True)
+    assert z_with_warmup.shape == (38, 4, 3)
+
+    # mostly just asserting that from_csv always succeeds
+    # in parsing latest cmdstan headers
+    fit_from_csv = from_csv(fit.runset.csv_files)
+    z_from_csv = fit_from_csv.stan_variable(var="z")
+    assert z_from_csv.shape == (20, 4, 3)
+    z_with_warmup_from_csv = fit.stan_variable(var="z", inc_warmup=True)
+    assert z_with_warmup_from_csv.shape == (38, 4, 3)
+
+
 @pytest.mark.order(before="test_no_xarray")
 def test_serialization(stanfile='bernoulli.stan'):
     # This test must before any test that uses the `without_import` context

{cmdstanpy-1.2.3 → cmdstanpy-1.2.4}/test/test_variational.py

@@ -15,6 +15,7 @@ import pytest
 from cmdstanpy.cmdstan_args import CmdStanArgs, VariationalArgs
 from cmdstanpy.model import CmdStanModel
 from cmdstanpy.stanfit import CmdStanVB, RunSet, from_csv
+from cmdstanpy.utils.cmdstan import cmdstan_version_before
 
 HERE = os.path.dirname(os.path.abspath(__file__))
 DATAFILES_PATH = os.path.join(HERE, 'data')
@@ -150,12 +151,12 @@ def test_variational_good() -> None:
         'mu[2]',
     )
     # fixed seed, id=1 by default will give known output values
-    assert variational.eta == 100
+    assert variational.eta == 10
     np.testing.assert_almost_equal(
-        variational.variational_params_dict['mu[1]'], 311.545, decimal=2
+        variational.variational_params_dict['mu[1]'], 302.142, decimal=2
     )
     np.testing.assert_almost_equal(
-        variational.variational_params_dict['mu[2]'], 532.801, decimal=2
+        variational.variational_params_dict['mu[2]'], 361.005, decimal=2
     )
     np.testing.assert_almost_equal(
         variational.variational_params_np[0],
@@ -177,7 +178,11 @@ def test_variational_eta_small() -> None:
         DATAFILES_PATH, 'variational', 'eta_should_be_small.stan'
     )
     model = CmdStanModel(stan_file=stan)
-    variational = model.variational(algorithm='meanfield', seed=12345)
+    if cmdstan_version_before(2, 35):
+        seed = 12345
+    else:
+        seed = 1234
+    variational = model.variational(algorithm='meanfield', seed=seed)
     assert variational.column_names == (
         'lp__',
         'log_p__',

cmdstanpy-1.2.3/cmdstanpy/_version.py

@@ -1,3 +0,0 @@
-"""PyPi Version"""
-
-__version__ = '1.2.3'