From 12dbdf789d3c10e2ca7e20aa033a03c2f7f107c4 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E8=88=AA=E5=AE=87?= <3364451258@qq.com>
Date: Thu, 28 May 2026 08:55:12 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20ERA5=20=E9=A2=84=E5=A4=84=E7=90=86?=
 =?UTF-8?q?=E5=AE=8C=E6=88=90=20=E2=80=94=20=E4=BF=AE=E5=A4=8D=20NetCDF=20?=
 =?UTF-8?q?=E5=BC=95=E6=93=8E=20(h5netcdf+dask)=EF=BC=8C=E7=BB=9F=E4=B8=80?=
 =?UTF-8?q?=20NPZ=20=E5=91=BD=E5=90=8D=EF=BC=8C109.6=E4=B8=87=E5=BA=8F?=
 =?UTF-8?q?=E5=88=97=E6=A0=B7=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pyproject.toml         | 1 +
 src/data/preprocess.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index a997dcc..3e3cab8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,6 +23,7 @@ dependencies = [
     "scipy>=1.11",
     "h5netcdf>=1.8.1",
     "h5py>=3.16.0",
+    "dask>=2026.3.0",
 ]
 
 [tool.uv.sources]
diff --git a/src/data/preprocess.py b/src/data/preprocess.py
index 01b8e76..20ab961 100644
--- a/src/data/preprocess.py
+++ b/src/data/preprocess.py
@@ -69,7 +69,7 @@ def load_era5_city(city: str) -> xr.Dataset:
     combined = xr.open_mfdataset(
         nc_files,
         combine="by_coords",
-        engine="netcdf4",
+        engine="h5netcdf",
         chunks=None,  # 小区域数据直接加载到内存
     )