...

Author: Borda

tests/__init__.py

@@ -17,7 +17,9 @@
 import numpy as np
 import torch
 
-from pytorch_lightning.utilities import _TORCH_LOWER_EQUAL_1_4, _TORCH_QUANTIZE_AVAILABLE
+logging.basicConfig(level=logging.ERROR)
+
+from pytorch_lightning.utilities import _TORCH_LOWER_EQUAL_1_4, _TORCH_QUANTIZE_AVAILABLE  # noqa: E402
 
 _TEST_ROOT = os.path.dirname(__file__)
 _PROJECT_ROOT = os.path.dirname(_TEST_ROOT)
@@ -36,10 +38,7 @@
 if not os.path.isdir(_TEMP_PATH):
     os.mkdir(_TEMP_PATH)
 
-logging.basicConfig(level=logging.ERROR)
-
 _MISS_QUANT_DEFAULT = 'fbgemm' not in torch.backends.quantized.supported_engines
-
 _SKIPIF_ARGS_PT_LE_1_4 = dict(condition=_TORCH_LOWER_EQUAL_1_4, reason="test pytorch > 1.4")
 _SKIPIF_ARGS_NO_GPU = dict(condition=not torch.cuda.is_available(), reason="test requires single-GPU machine")
 _SKIPIF_ARGS_NO_GPUS = dict(condition=torch.cuda.device_count() < 2, reason="test requires multi-GPU machine")