diff --git a/src/rechunk_data/_rechunk.py b/src/rechunk_data/_rechunk.py index 8d104f5908e38bd31f2d4bb0979f3bf33630433e..6e6a26ac0bb747800228a91d2a0577b4a5e4e3da 100644 --- a/src/rechunk_data/_rechunk.py +++ b/src/rechunk_data/_rechunk.py @@ -102,11 +102,7 @@ def _rechunk_dataset( logger.debug("Rechunking variable %s", var) chunks: Dict[int, Optional[str]] = {} for i, dim in enumerate(map(str, dset[var].dims)): - if ( - "lon" in dim.lower() - or "lat" in dim.lower() - or "bnds" in dim.lower() - ): + if "lon" in dim.lower() or "lat" in dim.lower() or "bnds" in dim.lower(): chunks[i] = None else: chunks[i] = "auto" @@ -124,14 +120,9 @@ def _rechunk_dataset( ) logger.debug("Settings encoding of variable %s", var) encoding[data_var] = { - str(k): v - for k, v in dset[var].encoding.items() - if str(k) in _keywords + str(k): v for k, v in dset[var].encoding.items() if str(k) in _keywords } - if ( - engine != "netcdf4" - or encoding[data_var].get("contiguous", False) is False - ): + if engine != "netcdf4" or encoding[data_var].get("contiguous", False) is False: encoding[data_var]["chunksizes"] = new_chunks return dset, encoding diff --git a/src/rechunk_data/tests/test_rechunk_netcdf.py b/src/rechunk_data/tests/test_rechunk_netcdf.py index 81d02ff2e1b0dd3f11e74df07ac54a1e93ac8fa9..3544d3d6138bbc1c9b4204c9b206ac88b41708b2 100644 --- a/src/rechunk_data/tests/test_rechunk_netcdf.py +++ b/src/rechunk_data/tests/test_rechunk_netcdf.py @@ -67,6 +67,5 @@ def test_wrong_or_format(small_chunk_data, caplog) -> None: def test_wrong_engine(small_chunk_data) -> None: - with pytest.raises(ValueError): rechunk_dataset(small_chunk_data, engine="foo")