earth2studio.data.fetch_data#

earth2studio.data.fetch_data(source, time, variable, lead_time=array([0], dtype='timedelta64[h]'), device='cpu', interp_to=None, interp_method='nearest')[source]#

Utility function to fetch data for models and load data on the target device. If desired, xarray interpolation/regridding in the spatial domain can be used by passing a target coordinate system via the optional interp_to argument.

Parameters:
  • source (DataSource) – The data source to fetch from

  • time (TimeArray) – Timestamps to return data for (UTC).

  • variable (VariableArray) – Strings or list of strings that refer to variables to return

  • lead_time (LeadTimeArray, optional) – Lead times to fetch for each provided time, by default np.array(np.timedelta64(0, “h”))

  • device (torch.device, optional) – Torch devive to load data tensor to, by default “cpu”

  • interp_to (CoordSystem, optional) – If provided, the fetched data will be interpolated to the coordinates specified by lat/lon arrays in this CoordSystem

  • interp_method (str) – Interpolation method to use with xarray (by default ‘nearest’)

Returns:

Tuple containing output tensor and coordinate OrderedDict

Return type:

tuple[torch.Tensor, CoordSystem]

Examples using earth2studio.data.fetch_data#

Statistical Inference

Statistical Inference

Tropical Cyclone Tracking

Tropical Cyclone Tracking

Running DLESyM Inference

Running DLESyM Inference

CBottle Data Generation and Infilling

CBottle Data Generation and Infilling

CBottle Super Resolution

CBottle Super Resolution