[Refactoring] Reordering imports in torch/utils/data/datapipes/iter/__init__.py (#61325)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61325
Test Plan: Imported from OSS
Reviewed By: ejguan
Differential Revision: D29588896
Pulled By: VitalyFedyunin
fbshipit-source-id: 8c0f3580f82083c43a590a18ecddb3e04ae93ca9
diff --git a/torch/utils/data/datapipes/iter/__init__.py b/torch/utils/data/datapipes/iter/__init__.py
index 0fb668b..b7718f8 100644
--- a/torch/utils/data/datapipes/iter/__init__.py
+++ b/torch/utils/data/datapipes/iter/__init__.py
@@ -1,37 +1,68 @@
-from torch.utils.data.datapipes.iter.listdirfiles import \
- ListDirFilesIterDataPipe as ListDirFiles
-from torch.utils.data.datapipes.iter.loadfilesfromdisk import \
- LoadFilesFromDiskIterDataPipe as LoadFilesFromDisk
-from torch.utils.data.datapipes.iter.readfilesfromtar import \
- ReadFilesFromTarIterDataPipe as ReadFilesFromTar
-from torch.utils.data.datapipes.iter.readfilesfromzip import \
- ReadFilesFromZipIterDataPipe as ReadFilesFromZip
-from torch.utils.data.datapipes.iter.routeddecoder import \
- RoutedDecoderIterDataPipe as RoutedDecoder
-from torch.utils.data.datapipes.iter.httpreader import \
- HTTPReaderIterDataPipe as HttpReader
-from torch.utils.data.datapipes.iter.tobytes import \
- ToBytesIterDataPipe as ToBytes
-from torch.utils.data.datapipes.iter.readlinesfromfile import \
- ReadLinesFromFileIterDataPipe as ReadLinesFromFile
+from torch.utils.data.datapipes.iter.callable import (
+ CollateIterDataPipe as Collate,
+ MapIterDataPipe as Map,
+ TransformsIterDataPipe as Transforms,
+)
+from torch.utils.data.datapipes.iter.combinatorics import (
+ SamplerIterDataPipe as Sampler,
+ ShuffleIterDataPipe as Shuffle,
+)
+from torch.utils.data.datapipes.iter.combining import (
+ ConcatIterDataPipe as Concat,
+ ZipIterDataPipe as Zip,
+)
+from torch.utils.data.datapipes.iter.grouping import (
+ BatchIterDataPipe as Batch,
+ BucketBatchIterDataPipe as BucketBatch,
+ GroupByKeyIterDataPipe as GroupByKey,
+)
+from torch.utils.data.datapipes.iter.httpreader import (
+ HTTPReaderIterDataPipe as HttpReader,
+)
+from torch.utils.data.datapipes.iter.listdirfiles import (
+ ListDirFilesIterDataPipe as ListDirFiles,
+)
+from torch.utils.data.datapipes.iter.loadfilesfromdisk import (
+ LoadFilesFromDiskIterDataPipe as LoadFilesFromDisk,
+)
+from torch.utils.data.datapipes.iter.readfilesfromtar import (
+ ReadFilesFromTarIterDataPipe as ReadFilesFromTar,
+)
+from torch.utils.data.datapipes.iter.readfilesfromzip import (
+ ReadFilesFromZipIterDataPipe as ReadFilesFromZip,
+)
+from torch.utils.data.datapipes.iter.readlinesfromfile import (
+ ReadLinesFromFileIterDataPipe as ReadLinesFromFile,
+)
+from torch.utils.data.datapipes.iter.routeddecoder import (
+ RoutedDecoderIterDataPipe as RoutedDecoder,
+)
+from torch.utils.data.datapipes.iter.selecting import (
+ FilterIterDataPipe as Filter,
+)
+from torch.utils.data.datapipes.iter.tobytes import (
+ ToBytesIterDataPipe as ToBytes,
+)
-# Functional DataPipe
-from torch.utils.data.datapipes.iter.callable import \
- (MapIterDataPipe as Map, CollateIterDataPipe as Collate,
- TransformsIterDataPipe as Transforms)
-from torch.utils.data.datapipes.iter.combining import \
- (ConcatIterDataPipe as Concat, ZipIterDataPipe as Zip)
-from torch.utils.data.datapipes.iter.combinatorics import \
- (SamplerIterDataPipe as Sampler, ShuffleIterDataPipe as Shuffle)
-from torch.utils.data.datapipes.iter.grouping import \
- (BatchIterDataPipe as Batch, BucketBatchIterDataPipe as BucketBatch,
- GroupByKeyIterDataPipe as GroupByKey)
-from torch.utils.data.datapipes.iter.selecting import \
- (FilterIterDataPipe as Filter)
+__all__ = ['Batch',
+ 'BucketBatch',
+ 'Collate',
+ 'Concat',
+ 'Filter',
+ 'GroupByKey',
+ 'HttpReader',
+ 'ListDirFiles',
+ 'LoadFilesFromDisk',
+ 'Map',
+ 'ReadFilesFromTar',
+ 'ReadFilesFromZip',
+ 'ReadLinesFromFile',
+ 'RoutedDecoder',
+ 'Sampler',
+ 'Shuffle',
+ 'ToBytes',
+ 'Transforms',
+ 'Zip']
-
-__all__ = ['ListDirFiles', 'LoadFilesFromDisk', 'ReadFilesFromTar',
- 'ReadFilesFromZip', 'RoutedDecoder', 'GroupByKey',
- 'Batch', 'BucketBatch', 'Collate', 'Concat', 'Filter', 'Map',
- 'Sampler', 'Shuffle', 'Transforms', 'Zip',
- 'HttpReader', 'ToBytes', 'ReadLinesFromFile']
+# Please keep this list sorted
+assert __all__ == sorted(__all__)
