fix: 统一 NPZ 命名格式,添加 NetCDF ZIP 解压工具和 h5netcdf/h5py 依赖
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user