Skip to content

Commit ac37895

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

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

tests/callbacks/test_progress_bar.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -482,14 +482,18 @@ def test_progress_bar_print_disabled(tqdm_write, mock_print, tmpdir):
482482
call("test_step"),
483483
])
484484
tqdm_write.assert_not_called()
485-
485+
486+
486487
def test_progbar_pickle():
487488
bar = ProgressBar()
488-
trainer = Trainer(fast_dev_run=True, callbacks=[bar],
489-
limit_train_batches=1,
490-
limit_val_batches=1,
491-
limit_test_batches=1,
492-
max_steps=1)
489+
trainer = Trainer(
490+
fast_dev_run=True,
491+
callbacks=[bar],
492+
limit_train_batches=1,
493+
limit_val_batches=1,
494+
limit_test_batches=1,
495+
max_steps=1
496+
)
493497
model = BoringModel()
494498
pickle.dumps(bar)
495499
trainer.fit(model)

0 commit comments

Comments
 (0)