File tree Expand file tree Collapse file tree 2 files changed +1
-6
lines changed
Expand file tree Collapse file tree 2 files changed +1
-6
lines changed Original file line number Diff line number Diff 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-
273268def 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 )
Original file line number Diff line number Diff 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" ):
You can’t perform that action at this time.
0 commit comments