from .webdataset_reader import SimpleImageDataset, PretoeknizedDataSetJSONL