[BE][Easy][8/19] enforce style for empty lines in import segments in test/[k-p]*/ (pytorch#129759)

See pytorch#129751 (comment). Most changes are auto-generated by linter.

You can review these PRs via:

```bash
git diff --ignore-all-space --ignore-blank-lines HEAD~1
```

Pull Request resolved: pytorch#129759
Approved by: https://github.com/justinchuby, https://github.com/ezyang

Author: XuehaiPan

test/lazy/test_debug_util.py

@@ -10,6 +10,7 @@
 import torch.nn as nn
 from torch.testing._internal.common_utils import IS_WINDOWS, run_tests, TestCase
 
+
 torch._lazy.ts_backend.init()
 
 

test/lazy/test_extract_compiled_graph.py

@@ -4,6 +4,7 @@
 
 from torch._lazy.ts_backend import init as init_ts_backend
 
+
 init_ts_backend()
 import copy
 import dis

test/lazy/test_functionalization.py

@@ -7,6 +7,7 @@
 import torch._lazy.ts_backend
 from torch.testing._internal.common_utils import run_tests, TestCase
 
+
 torch._lazy.ts_backend.init()
 
 NODE_TYPE_PATTERN = re.compile(r", NodeType=[^\n]+")

test/lazy/test_generator.py

@@ -5,6 +5,7 @@
 import torch._lazy.ts_backend
 from torch.testing._internal.common_utils import run_tests, skipIfTorchDynamo, TestCase
 
+
 torch._lazy.ts_backend.init()
 
 

test/lazy/test_meta_kernel.py

@@ -4,9 +4,9 @@
 import torch._lazy
 import torch._lazy.ts_backend
 from torch import float16, float32
-
 from torch.testing._internal.common_utils import TestCase
 
+
 torch._lazy.ts_backend.init()
 
 

test/lazy/test_reuse_ir.py

@@ -11,6 +11,7 @@
 import torch._lazy.ts_backend
 from torch.testing._internal.common_utils import IS_WINDOWS, run_tests, TestCase
 
+
 torch._lazy.ts_backend.init()
 torch._lazy.config.set_reuse_ir(True)
 

test/lazy/test_step_closures.py

@@ -7,6 +7,7 @@
 import torch._lazy.ts_backend
 from torch.testing._internal.common_utils import run_tests, TestCase
 
+
 torch._lazy.ts_backend.init()
 
 

test/mobile/custom_build/prepare_model.py

@@ -9,6 +9,7 @@
 
 import torch
 
+
 # Download and trace the model.
 model = models.mobilenet_v2(weights=models.MobileNet_V2_Weights.IMAGENET1K_V1)
 model.eval()

test/mobile/lightweight_dispatch/tests_setup.py

@@ -1,13 +1,13 @@
 import functools
 import os
 import shutil
-
 import sys
 from io import BytesIO
 
 import torch
 from torch.jit.mobile import _export_operator_list, _load_for_lite_interpreter
 
+
 _OPERATORS = set()
 _FILENAMES = []
 _MODELS = []

test/mobile/model_test/gen_test_model.py

@@ -29,12 +29,7 @@
     NNUtilsModule,
     NNVisionModule,
 )
-from quantization_ops import (
-    FusedQuantModule,
-    GeneralQuantModule,
-    # DynamicQuantModule,
-    StaticQuantModule,
-)
+from quantization_ops import FusedQuantModule, GeneralQuantModule, StaticQuantModule
 from sampling_ops import SamplingOpsModule
 from tensor_ops import (
     TensorCreationOpsModule,
@@ -52,6 +47,7 @@
 import torch
 from torch.jit.mobile import _load_for_lite_interpreter
 
+
 test_path_ios = "ios/TestApp/models/"
 test_path_android = "android/pytorch_android/src/androidTest/assets/"
 

test/mobile/model_test/update_production_ops.py

@@ -7,6 +7,7 @@
 
 import yaml
 
+
 root_operators = {}
 traced_operators = {}
 kernel_metadata = {}

test/mobile/test_bytecode.py

@@ -20,6 +20,7 @@
 )
 from torch.testing._internal.common_utils import run_tests, TestCase
 
+
 pytorch_test_dir = Path(__file__).resolve().parents[1]
 
 # script_module_v4.ptl and script_module_v5.ptl source code

test/mobile/test_lite_script_module.py

@@ -7,7 +7,6 @@
 
 import torch
 import torch.utils.bundled_inputs
-
 from torch.jit.mobile import _export_operator_list, _load_for_lite_interpreter
 from torch.testing import FileCheck
 from torch.testing._internal.common_quantization import (

test/mobile/test_lite_script_type.py

@@ -7,7 +7,6 @@
 
 import torch
 import torch.utils.bundled_inputs
-
 from torch.jit.mobile import _load_for_lite_interpreter
 from torch.testing._internal.common_utils import run_tests, TestCase
 

test/mobile/test_upgrader_codegen.py

@@ -6,9 +6,9 @@
 
 from torch.jit.generate_bytecode import generate_upgraders_bytecode
 from torch.testing._internal.common_utils import run_tests, TestCase
-
 from torchgen.operator_versions.gen_mobile_upgraders import sort_upgrader, write_cpp
 
+
 pytorch_caffe2_dir = Path(__file__).resolve().parents[2]
 
 

test/mobile/test_upgraders.py

@@ -6,10 +6,10 @@
 
 import torch
 import torch.utils.bundled_inputs
-
 from torch.jit.mobile import _load_for_lite_interpreter
 from torch.testing._internal.common_utils import run_tests, TestCase
 
+
 pytorch_test_dir = Path(__file__).resolve().parents[1]
 
 

test/nn/test_convolution.py

@@ -6,12 +6,10 @@
 from itertools import product
 
 import torch
-
 import torch.autograd.forward_ad as fwAD
 import torch.backends.cudnn as cudnn
 import torch.nn as nn
 import torch.nn.functional as F
-
 from torch.testing import make_tensor
 from torch.testing._internal.common_cuda import (
     TEST_CUDA,
@@ -62,6 +60,7 @@
     TEST_WITH_ROCM,
 )
 
+
 AMPERE_OR_ROCM = TEST_WITH_ROCM or tf32_is_not_fp32()
 
 

test/nn/test_embedding.py

@@ -19,7 +19,6 @@
     skipMeta,
     TEST_WITH_ROCM,
 )
-
 from torch.testing._internal.common_nn import NNTestCase
 from torch.testing._internal.common_utils import (
     _assertGradAndGradgradChecks,

test/nn/test_init.py

@@ -9,7 +9,6 @@
 import torch
 import torch.nn.functional as F
 import torch.nn.init as init
-
 from torch.testing._internal.common_utils import (
     run_tests,
     skipIfNoLapack,
@@ -19,6 +18,7 @@
     TestCase,
 )
 
+
 if TEST_SCIPY:
     from scipy import stats
 

test/nn/test_load_state_dict.py

@@ -19,6 +19,7 @@
 )
 from torch.utils._pytree import tree_map
 
+
 if TEST_NUMPY:
     import numpy as np
 

test/nn/test_module_hooks.py

@@ -7,7 +7,6 @@
 import weakref
 from collections import namedtuple, OrderedDict
 from copy import deepcopy
-
 from functools import partial
 from tempfile import NamedTemporaryFile
 from typing import Any, Dict, List, Tuple

test/nn/test_multihead_attention.py

@@ -6,7 +6,6 @@
 
 import torch
 import torch.nn as nn
-
 from torch.nn import MultiheadAttention
 from torch.testing._internal.common_device_type import (
     dtypes,
@@ -22,6 +21,7 @@
     TEST_WITH_CROSSREF,
 )
 
+
 if TEST_NUMPY:
     import numpy as np
 

test/nn/test_parametrization.py

@@ -4,7 +4,6 @@
 from itertools import product
 
 import torch
-
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.nn.init as init

test/nn/test_pooling.py

@@ -13,7 +13,6 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-
 from torch import inf, nan
 from torch.autograd import gradcheck, gradgradcheck
 from torch.testing import make_tensor

test/nn/test_pruning.py

@@ -4,7 +4,6 @@
 import unittest.mock as mock
 
 import torch
-
 import torch.nn as nn
 import torch.nn.utils.prune as prune
 from torch.testing._internal.common_nn import NNTestCase

test/onnx/dynamo/test_dynamo_with_onnxruntime_backend.py

@@ -20,10 +20,10 @@
     _OrtBackendOptions as OrtBackendOptions,
     ExportOptions,
 )
-
 from torch.testing._internal import common_utils
 from torch.testing._internal.common_utils import skipIfNNModuleInlined
 
+
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 import onnx_test_common
 

test/onnx/dynamo/test_exporter_api.py

@@ -13,7 +13,6 @@
     ProtobufONNXProgramSerializer,
     ResolvedExportOptions,
 )
-
 from torch.testing._internal import common_utils
 
 

test/onnx/dynamo/test_registry_dispatcher.py

@@ -21,6 +21,7 @@
 )
 from torch.testing._internal import common_utils
 
+
 # TODO: this can only be global. https://github.com/microsoft/onnxscript/issues/805
 TCustomFloat = TypeVar("TCustomFloat", bound=Union[FLOAT16, FLOAT, DOUBLE, BFLOAT16])
 

test/onnx/error_reproduction.py

@@ -17,6 +17,7 @@
 
 import torch
 
+
 _MISMATCH_MARKDOWN_TEMPLATE = """\
 ### Summary
 

test/onnx/internal/test_diagnostics.py

@@ -16,6 +16,7 @@
 from torch.onnx._internal.fx import diagnostics as fx_diagnostics
 from torch.testing._internal import common_utils, logging_utils
 
+
 if typing.TYPE_CHECKING:
     import unittest
 

test/onnx/model_defs/dcgan.py

@@ -1,6 +1,7 @@
 import torch
 import torch.nn as nn
 
+
 # configurable
 bsz = 64
 imgsz = 64

test/onnx/onnx_test_common.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import contextlib
-
 import copy
 import dataclasses
 import io
@@ -26,7 +25,6 @@
 )
 
 import numpy as np
-
 import onnxruntime
 import pytest
 import pytorch_test_common
@@ -39,6 +37,7 @@
 from torch.testing._internal.opinfo import core as opinfo_core
 from torch.types import Number
 
+
 _NumericType = Union[Number, torch.Tensor, np.ndarray]
 _ModelType = Union[torch.nn.Module, Callable, torch_export.ExportedProgram]
 _InputArgsType = Optional[

test/onnx/pytorch_test_common.py

@@ -18,6 +18,7 @@
 from torch.onnx._internal import diagnostics
 from torch.testing._internal import common_utils
 
+
 pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
 sys.path.insert(-1, pytorch_test_dir)
 

test/onnx/test_export_modes.py

@@ -11,6 +11,7 @@
 from torch.autograd import Variable
 from torch.onnx import OperatorExportTypes
 
+
 # Make the helper files in test/ importable
 pytorch_test_dir = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
 sys.path.append(pytorch_test_dir)

test/onnx/test_fx_op_consistency.py

@@ -50,9 +50,7 @@
 )
 
 import error_reproduction
-
 import onnx_test_common
-
 import parameterized
 import pytest
 import pytorch_test_common

test/onnx/test_fx_passes.py

@@ -4,7 +4,6 @@
 import torch
 import torch._dynamo
 import torch.fx
-
 from torch.onnx._internal.fx.passes import _utils as pass_utils
 from torch.testing._internal import common_utils
 

test/onnx/test_fx_to_onnx.py

@@ -2,13 +2,12 @@
 from __future__ import annotations
 
 import logging
-
 import tempfile
-
 from typing import Mapping, Tuple
 
 import onnx
 import onnx.inliner
+
 import pytorch_test_common
 import transformers  # type: ignore[import]
 

test/onnx/test_fx_to_onnx_decomp_skip.py

@@ -3,6 +3,7 @@
 
 import onnx
 import onnx.inliner
+
 import pytorch_test_common
 
 import torch