Skip to content

Commit d49cc7c

Browse files
authored
Update test_progress_bar.py
1 parent c594a50 commit d49cc7c

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/callbacks/test_progress_bar.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -485,9 +485,11 @@ def test_progress_bar_print_disabled(tqdm_write, mock_print, tmpdir):
485485

486486
def test_progbar_pickle():
487487
bar = ProgressBar()
488-
trainer = Trainer(fast_dev_run=True, callbacks=[bar, limit_train_batches=1,
489-
limit_val_batches=1,
490-
limit_test_batches=1, max_steps=1)
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)
491493
model = BoringModel()
492494
pickle.dumps(bar)
493495
trainer.fit(model)

0 commit comments

Comments
 (0)