Reformat test imports

Author: jvwilliams23

tests/test_basic.py

@@ -2,10 +2,10 @@
 import torch
 import torch.distributed as dist
 import torch.nn as nn
-from distconv import DCTensor, DistConvDDP, ParallelStrategy
-
 from utils import cleanup_parallel_strategy, fp32_allclose
 
+from distconv import DCTensor, DistConvDDP, ParallelStrategy
+
 
 @pytest.fixture(scope="module")
 def parallel_strategy(device: torch.device):

tests/test_convtranspose.py

@@ -1,11 +1,9 @@
 import pytest
 import torch
-import torch.distributed as dist
 import torch.nn as nn
-from distconv import DCTensor, DistConvDDP, ParallelStrategy
-
 from utils import cleanup_parallel_strategy, fp32_allclose
 
+from distconv import DCTensor, DistConvDDP, ParallelStrategy
 
 
 @pytest.fixture(scope="module")

tests/test_ddp_with_distconv.py

@@ -2,12 +2,12 @@
 import torch
 import torch.distributed as dist
 import torch.nn as nn
-from distconv import DCTensor, DistConvDDP, ParallelStrategy
 from torch.distributed.tensor import Replicate, Shard, distribute_tensor
 from torch.nn.parallel import DistributedDataParallel as DDP
-
 from utils import cleanup_parallel_strategy, fp32_allclose
 
+from distconv import DCTensor, DistConvDDP, ParallelStrategy
+
 
 @pytest.fixture(scope="module")
 def parallel_strategy(device: torch.device):

tests/test_periodic.py

@@ -4,10 +4,10 @@
 import torch
 import torch.distributed as dist
 import torch.nn as nn
-from distconv import DCTensor, DistConvDDP, ParallelStrategy
-
 from utils import cleanup_parallel_strategy, fp32_allclose
 
+from distconv import DCTensor, DistConvDDP, ParallelStrategy
+
 
 def generate_configs():
     configs = []

tests/test_strides.py

@@ -2,10 +2,10 @@
 import torch
 import torch.distributed as dist
 import torch.nn as nn
-from distconv import DCTensor, DistConvDDP, ParallelStrategy
-
 from utils import cleanup_parallel_strategy, fp32_allclose
 
+from distconv import DCTensor, DistConvDDP, ParallelStrategy
+
 
 @pytest.fixture(scope="module")
 def parallel_strategy(device: torch.device):