Graph Machine Learning
AnemoI
English
hcookie129 commited on
Commit
08286fc
·
1 Parent(s): 1b8913a

fix: Update anemoi-utils to 0.4.35.post3

Browse files
Files changed (3) hide show
  1. pyproject.toml +1 -1
  2. run_AIFS_v2.0.ipynb +2 -2
  3. uv.lock +10 -12
pyproject.toml CHANGED
@@ -12,7 +12,7 @@ dependencies = [
12
  "anemoi-graphs==0.6.4",
13
  "anemoi-models==0.9.3",
14
  "anemoi-transform==0.1.16.post2",
15
- "anemoi-utils==0.4.35",
16
  "earthkit-data<1",
17
  "earthkit-plots<1",
18
  "flash-attn==2.7.4",
 
12
  "anemoi-graphs==0.6.4",
13
  "anemoi-models==0.9.3",
14
  "anemoi-transform==0.1.16.post2",
15
+ "anemoi-utils==0.4.35.post3",
16
  "earthkit-data<1",
17
  "earthkit-plots<1",
18
  "flash-attn==2.7.4",
run_AIFS_v2.0.ipynb CHANGED
@@ -91,7 +91,7 @@
91
  },
92
  "outputs": [],
93
  "source": [
94
- "# !pip install -q anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3\n",
95
  "# !pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
96
  "# !pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
97
  "# !pip install -q flash-attn==2.7.4.post1"
@@ -123,7 +123,7 @@
123
  "metadata": {},
124
  "outputs": [],
125
  "source": [
126
- "# %pip install anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3\n",
127
  "# %pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
128
  "# %pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
129
  "# %pip install -q --force-reinstall --no-deps \"flash-attn @ https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl\""
 
91
  },
92
  "outputs": [],
93
  "source": [
94
+ "# !pip install -q anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3 anemoi-utils==0.4.35.post3\n",
95
  "# !pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
96
  "# !pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
97
  "# !pip install -q flash-attn==2.7.4.post1"
 
123
  "metadata": {},
124
  "outputs": [],
125
  "source": [
126
+ "# %pip install anemoi-inference[huggingface]==0.8.3 anemoi-models==0.9.3 anemoi-utils==0.4.35.post3\n",
127
  "# %pip install -q torch==2.7.0 torch-geometric==2.6.1\n",
128
  "# %pip install -q earthkit-regrid==0.5.1 ecmwf-opendata==0.3.29 'earthkit-data<1.0.0'\n",
129
  "# %pip install -q --force-reinstall --no-deps \"flash-attn @ https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl\""
uv.lock CHANGED
@@ -36,8 +36,8 @@ dependencies = [
36
  { name = "anemoi-utils", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
37
  { name = "earthkit-data", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
38
  { name = "earthkit-plots", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
39
- { name = "flash-attn", version = "2.8.3", source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl" }, marker = "platform_machine == 'aarch64' and sys_platform == 'linux'" },
40
- { name = "flash-attn", version = "2.8.3", source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }, marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" },
41
  { name = "torch", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
42
  { name = "torch-geometric", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
43
  ]
@@ -64,7 +64,7 @@ requires-dist = [
64
  { name = "anemoi-plugins-ecmwf-inference", extras = ["opendata"], marker = "extra == 'inference'", specifier = "==0.2.1" },
65
  { name = "anemoi-training", marker = "extra == 'training'", specifier = "==0.6.3" },
66
  { name = "anemoi-transform", specifier = "==0.1.16.post2" },
67
- { name = "anemoi-utils", specifier = "==0.4.35" },
68
  { name = "earthkit-data", specifier = "<1" },
69
  { name = "earthkit-plots", specifier = "<1" },
70
  { name = "earthkit-regrid", marker = "extra == 'inference'", specifier = "==0.5.1" },
@@ -294,7 +294,7 @@ wheels = [
294
 
295
  [[package]]
296
  name = "anemoi-utils"
297
- version = "0.4.35"
298
  source = { registry = "https://pypi.org/simple" }
299
  dependencies = [
300
  { name = "aniso8601", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
@@ -307,9 +307,9 @@ dependencies = [
307
  { name = "rich", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
308
  { name = "tqdm", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
309
  ]
310
- sdist = { url = "https://files.pythonhosted.org/packages/70/70/50e7bd93327bd610b0d808d017c5dc9b87a21a3f6172a449307ca1480cfe/anemoi_utils-0.4.35.tar.gz", hash = "sha256:c259c918e7ae9582c6f8cac3bb137fb1326759703d82334c7ec568c00a07afe5", size = 136476, upload-time = "2025-08-12T16:11:52.902Z" }
311
  wheels = [
312
- { url = "https://files.pythonhosted.org/packages/d4/d0/219f4f280745d7462bb534a1de9197afe1ab81e81ae2768e23569cb1cdce/anemoi_utils-0.4.35-py3-none-any.whl", hash = "sha256:b0c98359852231c3e18c5b90dd8a5ba278a106d527fa97d9b578fbf162b185a2", size = 91019, upload-time = "2025-08-12T16:11:51.43Z" },
313
  ]
314
 
315
  [package.optional-dependencies]
@@ -1108,11 +1108,10 @@ version = "2.8.3"
1108
  source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl" }
1109
  resolution-markers = [
1110
  "python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'",
1111
- "python_full_version < '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'",
1112
  ]
1113
  dependencies = [
1114
- { name = "einops", marker = "platform_machine == 'aarch64' and sys_platform == 'linux'" },
1115
- { name = "torch", marker = "platform_machine == 'aarch64' and sys_platform == 'linux'" },
1116
  ]
1117
  wheels = [
1118
  { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl", hash = "sha256:9dabca7ac35b5b62773318c0a2124605841df4ceba107a9a508691699b999bcc" },
@@ -1130,11 +1129,10 @@ version = "2.8.3"
1130
  source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }
1131
  resolution-markers = [
1132
  "python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'",
1133
- "python_full_version < '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'",
1134
  ]
1135
  dependencies = [
1136
- { name = "einops", marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" },
1137
- { name = "torch", marker = "platform_machine == 'x86_64' and sys_platform == 'linux'" },
1138
  ]
1139
  wheels = [
1140
  { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl", hash = "sha256:21c74313440453b914aa282a7bc78c8944dbe3b6c44989c9b36b64cdd345c933" },
 
36
  { name = "anemoi-utils", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
37
  { name = "earthkit-data", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
38
  { name = "earthkit-plots", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
39
+ { name = "flash-attn", version = "2.8.3", source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl" }, marker = "python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'" },
40
+ { name = "flash-attn", version = "2.8.3", source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }, marker = "python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'" },
41
  { name = "torch", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
42
  { name = "torch-geometric", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
43
  ]
 
64
  { name = "anemoi-plugins-ecmwf-inference", extras = ["opendata"], marker = "extra == 'inference'", specifier = "==0.2.1" },
65
  { name = "anemoi-training", marker = "extra == 'training'", specifier = "==0.6.3" },
66
  { name = "anemoi-transform", specifier = "==0.1.16.post2" },
67
+ { name = "anemoi-utils", specifier = "==0.4.35.post3" },
68
  { name = "earthkit-data", specifier = "<1" },
69
  { name = "earthkit-plots", specifier = "<1" },
70
  { name = "earthkit-regrid", marker = "extra == 'inference'", specifier = "==0.5.1" },
 
294
 
295
  [[package]]
296
  name = "anemoi-utils"
297
+ version = "0.4.35.post3"
298
  source = { registry = "https://pypi.org/simple" }
299
  dependencies = [
300
  { name = "aniso8601", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
 
307
  { name = "rich", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
308
  { name = "tqdm", marker = "(platform_machine == 'aarch64' and sys_platform == 'linux') or (platform_machine == 'x86_64' and sys_platform == 'linux')" },
309
  ]
310
+ sdist = { url = "https://files.pythonhosted.org/packages/be/4d/c31319293fb0a6fb0160d59e8e4b0a4233c5c5db95c8dedf6fe76582a505/anemoi_utils-0.4.35.post3.tar.gz", hash = "sha256:4ec1bc9e3843f98dd678d41db79572c3a73cc4752fb9bc76a5f74e3ed8ff494c", size = 142205, upload-time = "2026-05-18T14:10:53.654Z" }
311
  wheels = [
312
+ { url = "https://files.pythonhosted.org/packages/03/37/5b593014575e89354bf4e35d921b2f569b3c35f8ba5d8ec7142eb1ce2383/anemoi_utils-0.4.35.post3-py3-none-any.whl", hash = "sha256:8be900064d2d081ac5a712905f4cca3ec3895e5d3c2dde309d153e50e52afbe6", size = 92252, upload-time = "2026-05-18T14:10:52.311Z" },
313
  ]
314
 
315
  [package.optional-dependencies]
 
1108
  source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl" }
1109
  resolution-markers = [
1110
  "python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'",
 
1111
  ]
1112
  dependencies = [
1113
+ { name = "einops", marker = "python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'" },
1114
+ { name = "torch", marker = "python_full_version >= '3.12' and platform_machine == 'aarch64' and sys_platform == 'linux'" },
1115
  ]
1116
  wheels = [
1117
  { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_aarch64.whl", hash = "sha256:9dabca7ac35b5b62773318c0a2124605841df4ceba107a9a508691699b999bcc" },
 
1129
  source = { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl" }
1130
  resolution-markers = [
1131
  "python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'",
 
1132
  ]
1133
  dependencies = [
1134
+ { name = "einops", marker = "python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'" },
1135
+ { name = "torch", marker = "python_full_version >= '3.12' and platform_machine == 'x86_64' and sys_platform == 'linux'" },
1136
  ]
1137
  wheels = [
1138
  { url = "https://github.com/cathalobrien/get-flash-attn/releases/download/v0.1-alpha/flash_attn-2.8.3+cu12torch2.7cxx11abiFALSE-cp312-cp312-linux_x86_64.whl", hash = "sha256:21c74313440453b914aa282a7bc78c8944dbe3b6c44989c9b36b64cdd345c933" },