Skip to content

Commit b4617e8

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

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

pytorch_lightning/trainer/trainer.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
from pytorch_lightning.core.datamodule import LightningDataModule
3434
from pytorch_lightning.core.optimizer import LightningOptimizer
3535
from pytorch_lightning.loggers import LightningLoggerBase
36-
from pytorch_lightning.loggers.base import LoggerCollection, DummyLogger
36+
from pytorch_lightning.loggers.base import DummyLogger, LoggerCollection
3737
from pytorch_lightning.loggers.tensorboard import TensorBoardLogger
3838
from pytorch_lightning.loops import PredictionLoop, TrainingBatchLoop, TrainingEpochLoop
3939
from pytorch_lightning.loops.dataloader.evaluation_loop import EvaluationLoop
@@ -2165,11 +2165,12 @@ def terminate_on_nan(self, val: bool) -> None:
21652165
)
21662166
self._terminate_on_nan = val # : 212
21672167

2168+
21682169
def _determine_batch_limits(batches: Union[int, float], name: str) -> Union[int, float]:
2169-
if 0 <= batches <= 1:
2170-
return batches
2171-
if batches > 1 and batches % 1.0 == 0:
2172-
return int(batches)
2173-
raise MisconfigurationException(
2174-
f"You have passed invalid value {batches} for {name}, it has to be in [0.0, 1.0] or an int."
2175-
)
2170+
if 0 <= batches <= 1:
2171+
return batches
2172+
if batches > 1 and batches % 1.0 == 0:
2173+
return int(batches)
2174+
raise MisconfigurationException(
2175+
f"You have passed invalid value {batches} for {name}, it has to be in [0.0, 1.0] or an int."
2176+
)

0 commit comments

Comments
 (0)