Skip to content

Commit 67364e9

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

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/callbacks/test_stochastic_weight_avg.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424

2525
from pytorch_lightning import LightningModule, Trainer
2626
from pytorch_lightning.accelerators import Accelerator
27-
from pytorch_lightning.callbacks import StochasticWeightAveraging, ModelCheckpoint
27+
from pytorch_lightning.callbacks import ModelCheckpoint, StochasticWeightAveraging
2828
from pytorch_lightning.plugins import DDPSpawnPlugin
2929
from pytorch_lightning.trainer.connectors.data_connector import _PatchDataLoader
3030
from pytorch_lightning.utilities.exceptions import MisconfigurationException
@@ -380,7 +380,7 @@ def test_swa_load_best_checkpoint(tmpdir, batchnorm: bool, within_swa_epochs: bo
380380
max_epochs = 5
381381

382382
swa_callback = SwaTestCallback(swa_epoch_start=swa_start, swa_lrs=0.1, swa_validation=True)
383-
checkpoint_callback = ModelCheckpoint(monitor='val_loss', save_top_k=3, mode='min')
383+
checkpoint_callback = ModelCheckpoint(monitor="val_loss", save_top_k=3, mode="min")
384384

385385
trainer = Trainer(
386386
default_root_dir=tmpdir,

0 commit comments

Comments
 (0)