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

Improve history handling. #68

Closed nbren12 closed 11 months ago

nbren12 commented 11 months ago

Earth-2 MIP Pull Request

Description

Improve handling of history. Previously models with history (e.g. DLWP) could not be initialized with data sources without history. A work-around was to manually building a new data source, and made the notebooks/examples messy. Also, this was preventing using dlwp for ensemble inference with CDS inits for example.

I added support for history to the function get_initial_condition_for_model, which is used by all the inference routines.

Checklist

Dependencies

nbren12 commented 11 months ago

/blossom-ci

nbren12 commented 11 months ago

/blossom-ci

yairchn commented 11 months ago

LGTM