diff --git a/pytorch_lightning/strategies/fully_sharded.py b/pytorch_lightning/strategies/fully_sharded.py index 39b7ebe7a51dae..f99fa3462d4931 100644 --- a/pytorch_lightning/strategies/fully_sharded.py +++ b/pytorch_lightning/strategies/fully_sharded.py @@ -12,8 +12,8 @@ # See the License for the specific language governing permissions and # limitations under the License. import contextlib -from typing import Dict, Generator, List, Optional import logging +from typing import Dict, Generator, List, Optional import torch diff --git a/pytorch_lightning/trainer/connectors/checkpoint_connector.py b/pytorch_lightning/trainer/connectors/checkpoint_connector.py index 5af753b298b153..125548471b529b 100644 --- a/pytorch_lightning/trainer/connectors/checkpoint_connector.py +++ b/pytorch_lightning/trainer/connectors/checkpoint_connector.py @@ -12,9 +12,9 @@ # See the License for the specific language governing permissions and # limitations under the License. +import logging import os import re -import logging from typing import Any, Dict, Optional import torch diff --git a/tests/utilities/test_warnings.py b/tests/utilities/test_warnings.py index 5172bbaa1d3c6f..5262426a164819 100644 --- a/tests/utilities/test_warnings.py +++ b/tests/utilities/test_warnings.py @@ -53,6 +53,7 @@ # check that logging is properly configured import logging + from pytorch_lightning import _DETAIL root_logger = logging.getLogger()