From fb74fd9aded3013fb237a0b86b166384ea5f43d0 Mon Sep 17 00:00:00 2001 From: Philipp Weiler Date: Tue, 28 Apr 2026 10:50:59 -0400 Subject: [PATCH] fix: close store correctly --- src/fancypackage/io/_dask.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/fancypackage/io/_dask.py b/src/fancypackage/io/_dask.py index 9787084..bac8c83 100644 --- a/src/fancypackage/io/_dask.py +++ b/src/fancypackage/io/_dask.py @@ -1,9 +1,9 @@ from collections.abc import Generator from itertools import chain -from pathlib import Path import zarr from zarr import Group as ZarrGroup +from zarr.storage import StoreLike from anndata import AnnData from anndata.experimental import read_elem_lazy @@ -35,7 +35,7 @@ def _get_entries(group: ZarrGroup, level: str, entries: str | list[str] | None) def read_as_dask( - store: Path | str, + store: StoreLike, layers: str | list[str] | None = None, obsm_keys: str | list[str] | None = None, varm_keys: str | list[str] | None = None, @@ -88,6 +88,6 @@ def read_as_dask( for varm_key in varm_keys: adata.varm[varm_key] = read_elem_lazy(group[f"varm/{varm_key}"]) - store.close() + group.store.close() return adata