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