Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Oct 19, 2021
1 parent c37abbf commit 67364e9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/callbacks/test_stochastic_weight_avg.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@

from pytorch_lightning import LightningModule, Trainer
from pytorch_lightning.accelerators import Accelerator
from pytorch_lightning.callbacks import StochasticWeightAveraging, ModelCheckpoint
from pytorch_lightning.callbacks import ModelCheckpoint, StochasticWeightAveraging
from pytorch_lightning.plugins import DDPSpawnPlugin
from pytorch_lightning.trainer.connectors.data_connector import _PatchDataLoader
from pytorch_lightning.utilities.exceptions import MisconfigurationException
Expand Down Expand Up @@ -380,7 +380,7 @@ def test_swa_load_best_checkpoint(tmpdir, batchnorm: bool, within_swa_epochs: bo
max_epochs = 5

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

trainer = Trainer(
default_root_dir=tmpdir,
Expand Down

0 comments on commit 67364e9

Please sign in to comment.