14 lines
436 B
Python
14 lines
436 B
Python
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def test_removed_gpuaccelerator():
|
||
|
|
from pytorch_lightning.accelerators.gpu import GPUAccelerator
|
||
|
|
|
||
|
|
with pytest.raises(NotImplementedError, match="GPUAccelerator`.*no longer supported as of v1.9"):
|
||
|
|
GPUAccelerator()
|
||
|
|
|
||
|
|
from pytorch_lightning.accelerators import GPUAccelerator
|
||
|
|
|
||
|
|
with pytest.raises(NotImplementedError, match="GPUAccelerator`.*no longer supported as of v1.9"):
|
||
|
|
GPUAccelerator()
|