π[FEA]: make pure python function for earth2mip.time_collection
nbren12 opened this issue Β· comments
Noah D. Brenowitz commented
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