Skip to content

Commit 1eb83ac

Browse files
committed
add tpu mocks
1 parent 12f1908 commit 1eb83ac

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/tests_fabric/graveyard/test_tpu.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
("lightning.fabric.strategies.single_tpu", "SingleTPUStrategy"),
1212
],
1313
)
14-
def test_graveyard_single_tpu(import_path, name):
14+
def test_graveyard_single_tpu(import_path, name, tpu_available):
1515
module = import_module(import_path)
1616
cls = getattr(module, name)
1717
device = torch.device("cpu")
@@ -34,7 +34,7 @@ def test_graveyard_single_tpu(import_path, name):
3434
("lightning.fabric.plugins.precision.xlabf16", "XLABf16Precision"),
3535
],
3636
)
37-
def test_graveyard_no_device(import_path, name):
37+
def test_graveyard_no_device(import_path, name, tpu_available):
3838
module = import_module(import_path)
3939
cls = getattr(module, name)
4040
with pytest.deprecated_call(match="is deprecated"):

0 commit comments

Comments
 (0)