NVIDIA / earth2studio

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.

https://nvidia.github.io/earth2studio/

Apache License 2.0

73 stars 23 forks source link

CorrDiff import and example fix #81

Closed NickGeneva closed 2 months ago

NickGeneva commented 2 months ago

Earth2Studio Pull Request

Description

Fixing CorrDiff bare metal with an optional import for einops via pip install .[corrdiff].
Bug fix to corrdiff example
Improved examples to all create outputs dir if none exists.
Adding [parallel] install option for xarray to install Dask if not present.

Closes: https://github.com/NVIDIA/earth2studio/issues/62

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 2 months ago

/blossom-ci

NickGeneva commented 2 months ago

/blossom-ci