Skip to content

Commit d4bb25f

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

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

tests/tests_pytorch/callbacks/test_weight_averaging.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -426,9 +426,15 @@ def test_ema_weight_averaging_checkpoint_save_load(tmp_path):
426426
# Resume from checkpoint
427427
model2 = TestModel()
428428
callback2 = EMAWeightAveraging(decay=0.99, update_every_n_steps=2)
429-
import glob # should be at the top
430-
_train(model2, dataset, tmp_path, callback2,
431-
checkpoint_path=glob.glob((tmp_path / "checkpoints" / "*.ckpt").as_posix())[0])
429+
import glob # should be at the top
430+
431+
_train(
432+
model2,
433+
dataset,
434+
tmp_path,
435+
callback2,
436+
checkpoint_path=glob.glob((tmp_path / "checkpoints" / "*.ckpt").as_posix())[0],
437+
)
432438

433439
assert callback2._average_model is not None
434440

0 commit comments

Comments
 (0)