Fix typing issue

lucianopaz · lucianopaz · commit 134b40b3d2ae · 2025-05-14T14:45:47.000+02:00
diff --git a/pymc/backends/zarr.py b/pymc/backends/zarr.py
@@ -297,6 +297,7 @@ def completed_draws_and_divergences(self, chain_specific: bool = True) -> tuple[
         """
         # No need to iterate over ZarrChain instances because the zarr group is
         # shared between them
+        idx: int | slice
         if chain_specific:
             idx = self.chain
         else:
