stanio 0.2

Author: WardBrian

.github/workflows/main.yml

@@ -15,6 +15,11 @@ on:
         required: false
         default: ''
 
+# only run one copy per PR
+concurrency:
+  group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
+  cancel-in-progress: true
+
 jobs:
   get-cmdstan-version:
     # get the latest cmdstan version to use as part of the cache key

cmdstanpy/stanfit/gq.py

@@ -587,17 +587,14 @@ def stan_variable(
                 )
             elif isinstance(self.previous_fit, CmdStanMLE):
                 return np.atleast_1d(  # type: ignore
-                    np.asarray(
-                        self.previous_fit.stan_variable(
-                            var, inc_iterations=inc_warmup
-                        )
+                    self.previous_fit.stan_variable(
+                        var, inc_iterations=inc_warmup
                     )
                 )
             else:
                 return np.atleast_1d(  # type: ignore
-                    np.asarray(self.previous_fit.stan_variable(var))
+                    self.previous_fit.stan_variable(var)
                 )
-
         # is gq variable
         self._assemble_generated_quantities()
 

cmdstanpy/stanfit/metadata.py

@@ -39,12 +39,12 @@ def cmdstan_config(self) -> Dict[str, Any]:
         return copy.deepcopy(self._cmdstan_config)
 
     @property
-    def method_vars(self) -> Dict[str, stanio.Parameter]:
+    def method_vars(self) -> Dict[str, stanio.Variable]:
         """
         Method variable names always end in `__`, e.g. `lp__`.
         """
         return self._method_vars
 
     @property
-    def stan_vars(self) -> Dict[str, stanio.Parameter]:
+    def stan_vars(self) -> Dict[str, stanio.Variable]:
         return self._stan_vars

cmdstanpy/stanfit/mle.py

@@ -209,11 +209,11 @@ def stan_variable(
         if inc_iterations and self._save_iterations:
             data = self._all_iters
         else:
-            data = np.atleast_2d(self._mle)
+            data = self._mle
 
         try:
-            out: np.ndarray = (
-                self._metadata.stan_vars[var].extract_reshape(data).squeeze()
+            out: np.ndarray = self._metadata.stan_vars[var].extract_reshape(
+                data
             )
             return out
         except KeyError:

cmdstanpy/utils/data_munging.py

@@ -28,7 +28,7 @@ def flatten_chains(draws_array: np.ndarray) -> np.ndarray:
 
 def build_xarray_data(
     data: MutableMapping[Hashable, Tuple[Tuple[str, ...], np.ndarray]],
-    var: stanio.Parameter,
+    var: stanio.Variable,
     drawset: np.ndarray,
 ) -> None:
     """

requirements.txt

@@ -1,4 +1,4 @@
 pandas
 numpy>=1.21
 tqdm
-stanio~=0.1.0
+stanio~=0.2.0