Skip to content

Commit 498eb9e

Browse files
authored
Remove default chunks (#57)
Closes #45
1 parent 280f295 commit 498eb9e

File tree

2 files changed

+5
-9
lines changed

2 files changed

+5
-9
lines changed

tests/test_core.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def test_to_xarray_reference_file():
5050
with pytest.raises(ReferenceNotReachable):
5151
to_xarray(kerchunk_asset)
5252

53-
ds = to_xarray(kerchunk_asset, patch_url=pc.sign)
53+
ds = to_xarray(kerchunk_asset, patch_url=pc.sign, chunks={})
5454
assert not ds.lon.isnull().all(), "Coordinates should be populated"
5555

5656
for da in ds.data_vars.values():
@@ -69,7 +69,7 @@ def test_to_xarray_zarr():
6969
assert collection is not None
7070
zarr_asset = collection.assets["zarr-abfs"]
7171

72-
ds = to_xarray(zarr_asset)
72+
ds = to_xarray(zarr_asset, chunks={})
7373
for da in ds.data_vars.values():
7474
if da.ndim >= 2:
7575
assert hasattr(da.data, "dask"), da.name

xpystac/core.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
import functools
2-
from collections.abc import Callable, Mapping
2+
from collections.abc import Callable
33

44
import pystac
55
import xarray
@@ -71,7 +71,6 @@ def _(
7171

7272
mapper = fsspec.filesystem("reference", fo=refs).get_mapper()
7373
default_kwargs = {
74-
"chunks": {},
7574
"engine": "zarr",
7675
"consolidated": False,
7776
}
@@ -86,7 +85,6 @@ def _(
8685
allow_kerchunk: bool = True,
8786
**kwargs,
8887
) -> xarray.Dataset:
89-
default_kwargs: Mapping = {"chunks": {}}
9088
open_kwargs = obj.extra_fields.get("xarray:open_kwargs", {})
9189

9290
storage_options = obj.extra_fields.get("xarray:storage_options", None)
@@ -107,23 +105,21 @@ def _(
107105
refs = patch_url(refs)
108106

109107
default_kwargs = {
110-
**default_kwargs,
111108
"engine": "kerchunk",
112109
}
113110
return xarray.open_dataset(refs, **{**default_kwargs, **open_kwargs, **kwargs})
114111

115112
if obj.media_type == pystac.MediaType.COG:
116113
_import_optional_dependency("rioxarray")
117-
default_kwargs = {**default_kwargs, "engine": "rasterio"}
114+
default_kwargs = {"engine": "rasterio"}
118115
elif obj.media_type in ["application/vnd+zarr", "application/vnd.zarr"]:
119116
_import_optional_dependency("zarr")
120117
zarr_kwargs = {}
121118
if "zarr:consolidated" in obj.extra_fields:
122119
zarr_kwargs["consolidated"] = obj.extra_fields["zarr:consolidated"]
123120
if "zarr:zarr_format" in obj.extra_fields:
124121
zarr_kwargs["zarr_format"] = obj.extra_fields["zarr:zarr_format"]
125-
126-
default_kwargs = {**default_kwargs, **zarr_kwargs, "engine": "zarr"}
122+
default_kwargs = {**zarr_kwargs, "engine": "zarr"}
127123
elif obj.media_type == "application/vnd.zarr+icechunk":
128124
from xpystac._icechunk import read_icechunk
129125

0 commit comments

Comments
 (0)