Skip to content

Commit aac9127

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

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

src/pytorch_lightning/trainer/trainer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -601,7 +601,9 @@ def _init_debugging_flags(
601601
"Logging and checkpointing is suppressed."
602602
)
603603

604-
self.limit_train_batches: Union[int, float] = _determine_batch_limits(limit_train_batches, "limit_train_batches")
604+
self.limit_train_batches: Union[int, float] = _determine_batch_limits(
605+
limit_train_batches, "limit_train_batches"
606+
)
605607
self.limit_val_batches = _determine_batch_limits(limit_val_batches, "limit_val_batches")
606608
self.limit_test_batches = _determine_batch_limits(limit_test_batches, "limit_test_batches")
607609
self.limit_predict_batches = _determine_batch_limits(limit_predict_batches, "limit_predict_batches")

src/pytorch_lightning/tuner/batch_size_scaling.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,13 @@
1414
import logging
1515
import os
1616
import uuid
17-
from typing import Any, Dict, Union, List, Optional, Tuple
17+
from typing import Any, Dict, List, Optional, Tuple, Union
1818

1919
from torch.utils.data import DataLoader
2020

2121
import pytorch_lightning as pl
2222
from pytorch_lightning.callbacks.callback import Callback
2323
from pytorch_lightning.loggers.logger import DummyLogger, Logger
24-
2524
from pytorch_lightning.utilities.data import has_len_all_ranks
2625
from pytorch_lightning.utilities.exceptions import MisconfigurationException
2726
from pytorch_lightning.utilities.memory import garbage_collection_cuda, is_oom_error

0 commit comments

Comments
 (0)