From a1401d91a9a5a6333e8ec847166676d6bd4c856d Mon Sep 17 00:00:00 2001
From: k204230 <bergemann@dkrz.de>
Date: Sat, 30 Sep 2023 11:34:54 +0200
Subject: [PATCH] Fix linting issues
---
src/rechunk_data/_rechunk.py | 15 +++------------
src/rechunk_data/tests/test_rechunk_netcdf.py | 1 -
2 files changed, 3 insertions(+), 13 deletions(-)
diff --git a/src/rechunk_data/_rechunk.py b/src/rechunk_data/_rechunk.py
index 8d104f5..6e6a26a 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 81d02ff..3544d3d 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")
--
GitLab