Skip to content

Commit a7716d3

Browse files
committed
make it standalone
1 parent f56e1db commit a7716d3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tests_pytorch/trainer/properties/test_estimated_stepping_batches.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def on_train_start(self):
148148
assert self.trainer.estimated_stepping_batches == len(self.train_dataloader()) // 8
149149

150150

151-
@RunIf(tpu=True)
151+
@RunIf(tpu=True, standalone=True)
152152
def test_num_stepping_batches_with_tpu_multi():
153153
"""Test stepping batches with the TPU strategy across multiple devices."""
154154
trainer = Trainer(accelerator="tpu", devices=8, max_epochs=1)

0 commit comments

Comments
 (0)