Data Writers (pyvisgen.io.datawriters)#

Data writer submodule of pyvisgen.io. Data writers allow writing data simulated with pyvisgen to different formats. All data writers are subclasses of the abstract base clase DataWriter and support context manager protocol.

Reference/API#

pyvisgen.io.datawriters Module#

Classes#

DataWriter(output_path, dataset_type, *args, ...)

Abstract base class for data writers in pyvisgen.

FITSWriter(output_path, **kwargs)

FITS file writer for pyvisgen visibility datasets.

H5Writer(output_path, dataset_type, **kwargs)

HDF5 file writer for pyvisgen datasets.

PTWriter(output_path, dataset_type, ...)

DataWriter class for saving data in PyTorch (.pt) format.

WDSShardWriter(output_path, *, dataset_type, ...)

WebDataset file writer for pyvisgen datasets.

Class Inheritance Diagram#

Inheritance diagram of pyvisgen.io.datawriters.DataWriter, pyvisgen.io.datawriters.FITSWriter, pyvisgen.io.datawriters.H5Writer, pyvisgen.io.datawriters.PTWriter, pyvisgen.io.datawriters.WDSShardWriter