Skip to content

Checkpoint metadata patching #112

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
frazane merged 5 commits into from
Feb 19, 2026
Merged

Checkpoint metadata patching #112

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
ca7629f
metadata patching
frazane Feb 18, 2026
e12a752
Merge branch 'main' into feat/streamline-metadata-patching
frazane Feb 18, 2026
71ebe8d
update configs
frazane Feb 18, 2026
333e3fa
use anemoi-inference@main
frazane Feb 19, 2026
3362918
TOT_PREC template hourly step
frazane Feb 19, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 0 additions & 1 deletion config/forecasters-ich1-oper.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,6 @@ runs:
label: stage_C_icon_1km
steps: 0/120/6
config: resources/inference/configs/sgm-forecaster-global-ich1-oper.yaml
disable_local_eccodes_definitions: true
extra_requirements:
- earthkit-utils<0.2.0
- earthkit-data<0.19.0
Expand Down
2 changes: 0 additions & 2 deletions config/forecasters-ich1.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@ runs:
label: stage_C_icon_1km
steps: 0/120/6
config: resources/inference/configs/sgm-forecaster-global-ich1.yaml
disable_local_eccodes_definitions: true
extra_requirements:
- earthkit-utils<0.2.0
- earthkit-data<0.19.0
Expand All @@ -24,7 +23,6 @@ runs:
label: stage_C_icon_1km_hidden_11
steps: 0/120/6
config: resources/inference/configs/sgm-forecaster-global-ich1.yaml
disable_local_eccodes_definitions: true
extra_requirements:
- earthkit-utils<0.2.0
- earthkit-data<0.19.0
Expand Down
19 changes: 4 additions & 15 deletions resources/inference/configs/sgm-forecaster-global-ich1-oper.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,14 +18,9 @@ output:
path: grib/{date}{time:04}_{step:03}.grib
encoding:
typeOfGeneratingProcess: 2
centre: lssw
templates:
- file:
path: resources/icon-ch1-typeOfLevel=surface.grib
variables: [lsm, msl, sp, z, skt, tp]
- file:
path: resources/icon-ch1-typeOfLevel=heightAboveGround.grib
variables: [2t, 2d, 10u, 10v]
- file: resources/icon-ch1-typeOfLevel=isobaricInhPa.grib
samples: resources/templates_index_icon.yaml
post_processors:
- extract_mask: # removes global points
mask: "lam_0/cutout_mask"
Expand All @@ -34,6 +29,7 @@ output:
path: grib/ifs-{date}{time:04}_{step:03}.grib
encoding:
typeOfGeneratingProcess: 2
centre: ecmf
templates:
samples: resources/templates_index_ifs.yaml
post_processors:
Expand All @@ -44,11 +40,4 @@ output:
- assign_mask: # fill local/global overlapping points with nan
mask: "global/cutout_mask"

patch_metadata:
config:
dataloader:
test:
dataset:
cutout:
- dataset: /store_new/mch/msopr/ml/datasets/mch-ich1-1km-2024-2025-1h-pl13-ifsnames-v1.0.zarr
- dataset: /store_new/mch/msopr/ml/datasets/aifs-od-an-oper-0001-mars-n320-2016-2025-6h-v1-for-single-v2.zarr
patch_metadata: resources/sgm-ich1-oper-patch_metadata.yaml
17 changes: 2 additions & 15 deletions resources/inference/configs/sgm-forecaster-global-ich1.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,13 +19,7 @@ output:
encoding:
typeOfGeneratingProcess: 2
templates:
- file:
path: resources/icon-ch1-typeOfLevel=surface.grib
variables: [lsm, msl, sp, z, skt, tp]
- file:
path: resources/icon-ch1-typeOfLevel=heightAboveGround.grib
variables: [2t, 2d, 10u, 10v]
- file: resources/icon-ch1-typeOfLevel=isobaricInhPa.grib
samples: resources/templates_index_icon.yaml
post_processors:
- extract_mask: # removes global points
mask: "lam_0/cutout_mask"
Expand All @@ -44,11 +38,4 @@ output:
- assign_mask: # fill local/global overlapping points with nan
mask: "global/cutout_mask"

patch_metadata:
config:
dataloader:
test:
dataset:
cutout:
- dataset: /store_new/mch/msopr/ml/datasets/mch-realch1-fdb-1km-2005-2025-1h-pl13-ifsnames-v1.0.zarr
- dataset: /store_new/mch/msopr/ml/datasets/aifs-ea-an-oper-0001-mars-n320-1979-2024-6h-v1-for-single-v2.zarr
patch_metadata: resources/sgm-ich1-patch_metadata.yaml
Loading
Loading