Sort imports of test_datapipe.py (#61312)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61312
Sorting according to isort output. Alphabetically ordered one per line imports help merging.
Test Plan: Imported from OSS
Reviewed By: ejguan
Differential Revision: D29588833
Pulled By: VitalyFedyunin
fbshipit-source-id: 4c80c3086132b50894e734ad6c5799d78d689e42
diff --git a/test/test_datapipe.py b/test/test_datapipe.py
index 145513b..f335e1a 100644
--- a/test/test_datapipe.py
+++ b/test/test_datapipe.py
@@ -1,39 +1,57 @@
+import http.server
import itertools
-import numpy as np
import os
import os.path
import pickle
import random
+import socketserver
import sys
import tarfile
import tempfile
-import warnings
-import zipfile
-
-import unittest
-from unittest import skipIf
-from typing import (
- Any, Awaitable, Dict, Generic, Iterator, List, NamedTuple, Optional, Tuple,
- Type, TypeVar, Set, Union)
-import http.server
-import socketserver
import threading
import time
+import unittest
+import warnings
+import zipfile
from functools import partial
+from typing import (
+ Any,
+ Awaitable,
+ Dict,
+ Generic,
+ Iterator,
+ List,
+ NamedTuple,
+ Optional,
+ Set,
+ Tuple,
+ Type,
+ TypeVar,
+ Union,
+)
+from unittest import skipIf
+
+import numpy as np
import torch
import torch.nn as nn
+import torch.utils.data.backward_compatibility
import torch.utils.data.datapipes as dp
import torch.utils.data.graph
import torch.utils.data.sharding
-import torch.utils.data.backward_compatibility
-
-from torch.testing._internal.common_utils import (TestCase, run_tests)
+from torch.testing._internal.common_utils import TestCase, run_tests
from torch.utils.data import (
- IterDataPipe, MapDataPipe, RandomSampler, DataLoader,
- argument_validation, runtime_validation_disabled, runtime_validation)
+ DataLoader,
+ IterDataPipe,
+ MapDataPipe,
+ RandomSampler,
+ argument_validation,
+ runtime_validation,
+ runtime_validation_disabled,
+)
from torch.utils.data.datapipes.utils.decoder import (
- basichandlers as decoder_basichandlers)
+ basichandlers as decoder_basichandlers,
+)
try:
import torchvision.transforms
@@ -126,7 +144,10 @@
def test_loadfilesfromdisk_iterable_datapipe(self):
# test import datapipe class directly
- from torch.utils.data.datapipes.iter import ListDirFiles, LoadFilesFromDisk
+ from torch.utils.data.datapipes.iter import (
+ ListDirFiles,
+ LoadFilesFromDisk,
+ )
temp_dir = self.temp_dir.name
datapipe1 = ListDirFiles(temp_dir, '')
