Skip to content
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.

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

Add StormCast model #164

Merged
merged 24 commits into from
Dec 20, 2024
Merged

Conversation

pzharrington
Copy link
Collaborator

@pzharrington pzharrington commented Dec 13, 2024

Earth2Studio Pull Request

Description

Adds the StormCast model as a prognostic model. Due to the use of curvilinear coordinates in HRRR data (lambert conformal projection) and conditioning data this incurs some minor changes elsewhere; mainly the ability to perform xarray interpolation to a desired set of coordinates has been added to fetch_data.

Checklist

  • I am familiar with the Contributing Guidelines.
  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.
  • The CHANGELOG.md is up to date with these changes.
  • An issue is linked to this pull request.

Dependencies

The StormCast model depends on newly added material in modulus which has not yet been released, so the toml needs an update.

@pzharrington
Copy link
Collaborator Author

/blossom-ci

earth2studio/models/batch.py Outdated Show resolved Hide resolved
earth2studio/models/px/__init__.py Show resolved Hide resolved
earth2studio/models/px/stormcast.py Outdated Show resolved Hide resolved
@pzharrington
Copy link
Collaborator Author

/blossom-ci

@dallasfoster
Copy link
Collaborator

Looks really good, just the few minor comments.

earth2studio/data/utils.py Show resolved Hide resolved
earth2studio/data/utils.py Show resolved Hide resolved
CHANGELOG.md Outdated Show resolved Hide resolved
@NickGeneva
Copy link
Collaborator

Oh also add this to the RST file:
https://github.com/NVIDIA/earth2studio/blob/main/docs/modules/models.rst

So it gets build into the docs 👍

Copy link
Collaborator

@NickGeneva NickGeneva left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Approving to unblock, looks really good.

@dallasfoster
Copy link
Collaborator

/blossom-ci

@pzharrington
Copy link
Collaborator Author

/blossom-ci

1 similar comment
@pzharrington
Copy link
Collaborator Author

/blossom-ci

@pzharrington pzharrington merged commit 4000de1 into NVIDIA:main Dec 20, 2024
1 check passed
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.

3 participants