Fix typing issue

lucianopaz · lucianopaz · commit 1b0a1626729e · 2025-02-21T11:26:09.000+01:00
diff --git a/pymc/backends/zarr.py b/pymc/backends/zarr.py
@@ -305,6 +305,7 @@ def completed_draws_and_divergences(self, chain_specific: bool = False) -> 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:
