Skip to content

Commit f5586c7

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 97eabcf commit f5586c7

File tree

2 files changed

+6
-7
lines changed

2 files changed

+6
-7
lines changed

pytorch_lightning/loops/batch/training_batch_loop.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,18 +27,17 @@
2727
from pytorch_lightning.core.optimizer import LightningOptimizer
2828
from pytorch_lightning.loops.base import Loop
2929
from pytorch_lightning.loops.closure import Closure, ClosureResult
30+
from pytorch_lightning.loops.optimizer.optimizer_loop import OptimizerLoop
3031
from pytorch_lightning.loops.utilities import (
3132
_check_training_step_output,
3233
_process_training_step_output,
3334
check_finite_loss,
3435
)
35-
from pytorch_lightning.loops.optimizer.optimizer_loop import OptimizerLoop
3636
from pytorch_lightning.plugins import ParallelPlugin
3737
from pytorch_lightning.trainer.progress import OptimizationProgress
3838
from pytorch_lightning.trainer.supporters import TensorRunningAccum
3939
from pytorch_lightning.utilities import AMPType, AttributeDict, DeviceType, grad_norm
4040
from pytorch_lightning.utilities.exceptions import MisconfigurationException
41-
4241
from pytorch_lightning.utilities.signature_utils import is_param_in_hook_signature
4342
from pytorch_lightning.utilities.types import STEP_OUTPUT
4443
from pytorch_lightning.utilities.warnings import WarningCache

pytorch_lightning/loops/optimizer/optimizer_loop.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,25 +2,25 @@
22
from contextlib import contextmanager
33
from copy import copy
44
from functools import partial
5-
from typing import Any, Dict, Optional, Callable, Generator, Sequence, List
5+
from typing import Any, Callable, Dict, Generator, List, Optional, Sequence
66

77
import torch
88
from torch import Tensor
99
from torch.optim import Optimizer
1010

11-
from pytorch_lightning.utilities.imports import _TPU_AVAILABLE
1211
from pytorch_lightning.core.optimizer import LightningOptimizer
1312
from pytorch_lightning.loops import Loop
14-
from pytorch_lightning.loops.closure import ClosureResult, Closure
13+
from pytorch_lightning.loops.closure import Closure, ClosureResult
1514
from pytorch_lightning.loops.utilities import (
16-
check_finite_loss,
1715
_check_training_step_output,
1816
_process_training_step_output,
17+
check_finite_loss,
1918
)
2019
from pytorch_lightning.plugins import ParallelPlugin
2120
from pytorch_lightning.trainer.progress import OptimizationProgress
22-
from pytorch_lightning.utilities import AttributeDict, AMPType, DeviceType, grad_norm
21+
from pytorch_lightning.utilities import AMPType, AttributeDict, DeviceType, grad_norm
2322
from pytorch_lightning.utilities.exceptions import MisconfigurationException
23+
from pytorch_lightning.utilities.imports import _TPU_AVAILABLE
2424

2525

2626
class OptimizerLoop(Loop):

0 commit comments

Comments
 (0)