traveler.skims.omx.OmxSkims.to_jax#

OmxSkims.to_jax(name='SKIMS', cls=JaxDataset)[source]#

Create a JAX dataset from the loaded skim arrays.

Parameters:

  • name (str or None, optional) – Dataset name stored in metadata.

  • cls (type, optional) – JaxDataset subclass to instantiate.

Returns:

Dataset containing the loaded skims and time-period metadata.

Return type:

JaxDataset