Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli committed Apr 29, 2024
1 parent 1c7d7c8 commit 13bdb79
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -152,6 +152,6 @@ def on_train_start(self):
@mock.patch.dict(os.environ, os.environ.copy(), clear=True)
def test_num_stepping_batches_with_tpu_multi():
"""Test stepping batches with the TPU strategy across multiple devices."""
trainer = Trainer(accelerator="tpu", devices="auto", max_epochs=1)
trainer = Trainer(accelerator="tpu", devices="auto", max_epochs=1, logger=False, enable_checkpointing=False)
model = MultiprocessModel()
trainer.fit(model)

0 comments on commit 13bdb79

Please sign in to comment.