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.

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

πŸš€[FEA]: make pure python function for earth2mip.time_collection

nbren12 opened this issue Β· comments

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Critical (currently preventing usage)

Please provide a clear description of problem you would like to solve.

The command line interface of earth2mip.time_collection is a bit hard to use. The model entry-point cannot be specified. It requires a file "config.json". It will be easier to make a python function that can be imported as needed rather than through the command line.

def run_inference_for_times(
     time_loop: TimeLoop, 
     data_source: DataSource
     initial_times: list[datetime.datetime], 
     config: earth2mip.schema.EnsembleRun,
     postprocessing_options,
     shard: int = 0,
     n_shards: int = 1):
     ...

Describe any alternatives you have considered

No response