NVIDIA / earth2mip

Earth-2 Model Intercomparison Project (MIP) is a python framework that enables climate researchers and scientists to inter-compare AI models for weather and climate.

https://nvidia.github.io/earth2mip/

Apache License 2.0

187 stars 41 forks source link

Feat: Adding NGC downloading of diagnostic models #96

Closed NickGeneva closed 10 months ago

NickGeneva commented 10 months ago

Earth-2 MIP Pull Request

Description

Adding the NGC model registry for diagnostic models. Enable users to auto download diagnostics using the correct e2mip:// url. Its built into the diagnostic load_package functions as the default. Unlike the other models, all diagnostics are stored in one model registry making access generally easier.
Added a markdown doc because I hate RST... docs need pandoc anyway for converting the Jupytext.
Also a few bug fixes: registry download when folder doesnt exist, softmax climatenet output
Tested both precip and Climatenet with auto download

Closes: https://github.com/NVIDIA/earth2mip/issues/90

test

test (1)

Checklist

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

Dependencies

NickGeneva commented 10 months ago

/blossom-ci

NickGeneva commented 10 months ago

/blossom-ci

NickGeneva commented 10 months ago

/blossom-ci

NickGeneva commented 10 months ago

/blossom-ci