Skip to content

Commit 8c59ba4

Browse files
committed
avoid leaking env vars
1 parent f56e1db commit 8c59ba4

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/tests_pytorch/trainer/properties/test_estimated_stepping_batches.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,7 @@ def on_train_start(self):
149149

150150

151151
@RunIf(tpu=True)
152+
@mock.patch.dict(os.environ, os.environ.copy(), clear=True)
152153
def test_num_stepping_batches_with_tpu_multi():
153154
"""Test stepping batches with the TPU strategy across multiple devices."""
154155
trainer = Trainer(accelerator="tpu", devices=8, max_epochs=1)

0 commit comments

Comments
 (0)