caustics.data package

Submodules

caustics.data.hdf5dataset module

class caustics.data.hdf5dataset.HDF5Dataset(path: str, keys: List[str], device: device = device(type='cpu'), dtypes: Dict[str, dtype] | dtype = torch.float32)

Bases: Dataset

Light-weight HDF5 dataset that reads all the data into tensors. Assumes all groups in dataset have the same length.

caustics.data.illustris_kappa module

class caustics.data.illustris_kappa.IllustrisKappaDataset(path: str, device: device = device(type='cpu'), dtype: dtype = torch.float32)

Bases: object

caustics.data.probes module

class caustics.data.probes.PROBESDataset(path: str, device: device = device(type='cpu'), dtype: dtype = torch.float32)

Bases: object

Module contents