fix ci

.azure-pipelines/azure-pipelines-benchmarks.yml

@@ -29,7 +29,7 @@ jobs:
       python -m pip install wheel
       python -m pip install --no-cache-dir -r requirements.txt
       python -m pip install --no-cache-dir -r requirements-optional.txt
-      python -m pip install asv!=0.6.2
+      python -m pip install asv==0.6.1
     displayName: 'Install requirements'
 
   - script: |

arviz/data/inference_data.py

@@ -254,8 +254,7 @@ def __len__(self) -> int:
 
     def __iter__(self) -> Iterator[str]:
         """Iterate over groups in InferenceData object."""
-        for group in self._groups_all:
-            yield group
+        yield from self._groups_all
 
     def __contains__(self, key: object) -> bool:
         """Return True if the named item is present, and False otherwise."""

arviz/data/io_pystan.py

@@ -676,8 +676,7 @@ def get_draws(fit, variables=None, ignore=None, warmup=False, dtypes=None):
         for item in par_keys:
             _, shape = item.replace("]", "").split("[")
             shape_idx_min = min(int(shape_value) for shape_value in shape.split(","))
-            if shape_idx_min < shift:
-                shift = shape_idx_min
+            shift = min(shift, shape_idx_min)
         # If shift is higher than 1, this will probably mean that Stan
         # has implemented sparse structure (saves only non-zero parts),
         # but let's hope that dims are still corresponding to the full shape