1from torch.utils.data.datapipes.iter.callable import ( 2 CollatorIterDataPipe as Collator, 3 MapperIterDataPipe as Mapper, 4) 5from torch.utils.data.datapipes.iter.combinatorics import ( 6 SamplerIterDataPipe as Sampler, 7 ShufflerIterDataPipe as Shuffler, 8) 9from torch.utils.data.datapipes.iter.combining import ( 10 ConcaterIterDataPipe as Concater, 11 DemultiplexerIterDataPipe as Demultiplexer, 12 ForkerIterDataPipe as Forker, 13 MultiplexerIterDataPipe as Multiplexer, 14 ZipperIterDataPipe as Zipper, 15) 16from torch.utils.data.datapipes.iter.filelister import ( 17 FileListerIterDataPipe as FileLister, 18) 19from torch.utils.data.datapipes.iter.fileopener import ( 20 FileOpenerIterDataPipe as FileOpener, 21) 22from torch.utils.data.datapipes.iter.grouping import ( 23 BatcherIterDataPipe as Batcher, 24 GrouperIterDataPipe as Grouper, 25 UnBatcherIterDataPipe as UnBatcher, 26) 27from torch.utils.data.datapipes.iter.routeddecoder import ( 28 RoutedDecoderIterDataPipe as RoutedDecoder, 29) 30from torch.utils.data.datapipes.iter.selecting import FilterIterDataPipe as Filter 31from torch.utils.data.datapipes.iter.sharding import ( 32 ShardingFilterIterDataPipe as ShardingFilter, 33) 34from torch.utils.data.datapipes.iter.streamreader import ( 35 StreamReaderIterDataPipe as StreamReader, 36) 37from torch.utils.data.datapipes.iter.utils import ( 38 IterableWrapperIterDataPipe as IterableWrapper, 39) 40 41 42__all__ = [ 43 "Batcher", 44 "Collator", 45 "Concater", 46 "Demultiplexer", 47 "FileLister", 48 "FileOpener", 49 "Filter", 50 "Forker", 51 "Grouper", 52 "IterableWrapper", 53 "Mapper", 54 "Multiplexer", 55 "RoutedDecoder", 56 "Sampler", 57 "ShardingFilter", 58 "Shuffler", 59 "StreamReader", 60 "UnBatcher", 61 "Zipper", 62] 63 64# Please keep this list sorted 65assert __all__ == sorted(__all__) 66