Skip to content

Allow for variable renaming in coupled modeling #1240

Open
Arcomano1234 wants to merge 2 commits into
mainfrom
feature/renaming-vars-coupled-modeling
Open

Allow for variable renaming in coupled modeling #1240
Arcomano1234 wants to merge 2 commits into
mainfrom
feature/renaming-vars-coupled-modeling

Conversation

@Arcomano1234

@Arcomano1234 Arcomano1234 commented Jun 8, 2026

Copy link
Copy Markdown
Contributor

Currently, we are attempting to mix and match an atmospheric model trained on ERA5 and an ocean model trained on UFS replay dataset. The problem is in ERA5 wind stresses are called eastward_surface_stress while in CM4 and UFS replay they are called eastward_surface_wind_stress. Given that we will not be using the atmospheric variables to train a new ACE version for the foreseeable future we need a way to allow variable renaming. Similar to how it was done for HiRO-ACE in the downscaling model, this PR adds an optional stepper arg atmosphere_output_rename that will remap the user specified atmospheric variables to be renamed when the ocean model is grabbing forcing variables. Can also be used during inference and standalone eval.

Changes:

  • add atmosphere_output_rename to coupled stepper that can rename atmospheric forcing variables for the ocean model

  • Tests added

@Arcomano1234 Arcomano1234 changed the title allow for renaming of forcing variables for the ocean that come from … Allow for variable renaming in coupled modeling Jun 8, 2026
@oliverwm1

Copy link
Copy Markdown
Collaborator

I think this would be a reasonable feature to support. Although in my experience there are always a lot of tricky edge cases. Previously we had support for renaming variables within data loading, but we removed this due to the complexity of supported merged datasets: https://github.com/ai2cm/full-model/pull/1511. Keeping it confined to the coupled stepper for now at least limits surface area.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants