fix: 统一 NPZ 命名格式,添加 NetCDF ZIP 解压工具和 h5netcdf/h5py 依赖

This commit is contained in:
2026-05-28 08:26:18 +08:00
parent 07468266b4
commit a2c9ba4863
3 changed files with 66 additions and 2 deletions
+14
View File
@@ -10,6 +10,8 @@ dependencies = [
"netcdf4>=1.6",
"cdsapi>=0.7",
"torch>=2.1",
"torchvision>=0.19",
"torchaudio>=2.5",
"pytorch-lightning>=2.1",
"xgboost>=2.0",
"scikit-learn>=1.3",
@@ -19,4 +21,16 @@ dependencies = [
"jupyter>=1.0",
"tqdm>=4.66",
"scipy>=1.11",
"h5netcdf>=1.8.1",
"h5py>=3.16.0",
]
[tool.uv.sources]
torch = { index = "pytorch-cu126" }
torchvision = { index = "pytorch-cu126" }
torchaudio = { index = "pytorch-cu126" }
[[tool.uv.index]]
name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu126"
explicit = true