Skip to content

Commit ecb8e00

Browse files
committed
update
1 parent 859cbfd commit ecb8e00

File tree

2 files changed

+1
-6
lines changed

2 files changed

+1
-6
lines changed

tests/tests_pytorch/conftest.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -265,11 +265,6 @@ def xla_available(monkeypatch: pytest.MonkeyPatch) -> None:
265265
mock_xla_available(monkeypatch)
266266

267267

268-
@pytest.fixture
269-
def xla_not_available(monkeypatch: pytest.MonkeyPatch) -> None:
270-
mock_xla_available(monkeypatch, False)
271-
272-
273268
def mock_tpu_available(monkeypatch: pytest.MonkeyPatch, value: bool = True) -> None:
274269
mock_xla_available(monkeypatch, value)
275270
monkeypatch.setattr(lightning.fabric.accelerators.xla.XLAAccelerator, "is_available", lambda: value)

tests/tests_pytorch/graveyard/test_tpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def test_graveyard_single_tpu(import_path, name):
3535
("lightning.pytorch.plugins.precision.xlabf16", "XLABf16PrecisionPlugin"),
3636
],
3737
)
38-
def test_graveyard_no_device(import_path, name, xla_not_available):
38+
def test_graveyard_no_device(import_path, name):
3939
module = import_module(import_path)
4040
cls = getattr(module, name)
4141
with pytest.deprecated_call(match="is deprecated"), pytest.raises(ModuleNotFoundError, match="torch_xla"):

0 commit comments

Comments
 (0)