search

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

Add and update output coordinate handling #63

Closed dallasfoster closed 3 months ago

dallasfoster commented 3 months ago

Earth2Studio Pull Request

Description

This PR changes the behavior of output coordinate handling in px and dx models, as well as in statistics. The intent of this PR is to make sure that each method can specify what the output coordinate system is for a given input coordinate system.

For all models the output_coords utility function should provide the resulting coordinates after using the associated __call__ method for each model. For prognostic models this method, similar to the __call__ method, may not necessarily be run autoregressively and there is not equivalent iterator method for computing output coordinates.

Checklist

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci

dallasfoster commented 3 months ago

/blossom-ci