earth2studio.run.diagnostic#
- earth2studio.run.diagnostic(time, nsteps, prognostic, diagnostic, data, io, output_coords={}, device=None)[source]#
Built in diagnostic workflow. This workflow creates a determinstic inference pipeline that couples a prognostic model with a diagnostic model.
- Parameters:
time (list[str] | list[datetime] | list[np.datetime64]) – List of string, datetimes or np.datetime64
nsteps (int) – Number of forecast steps
prognostic (PrognosticModel) – Prognostic model
diagnostic (DiagnosticModel) – Diagnostic model, must be on same coordinate axis as prognostic
data (DataSource) – Data source
io (IOBackend) – IO object
output_coords (CoordSystem, optional) – IO output coordinate system override, by default OrderedDict({})
device (torch.device, optional) – Device to run inference on, by default None
- Returns:
Output IO object
- Return type:
IOBackend
