24 lines
869 B
Python
24 lines
869 B
Python
|
import pytest
|
||
|
|
|
||
|
|
|
||
|
|
def test_lightningCLI_old_module_removal():
|
||
|
|
from pytorch_lightning.utilities.cli import LightningCLI
|
||
|
|
|
||
|
|
with pytest.raises(NotImplementedError, match=r"LightningCLI.*no longer supported as of v1.9"):
|
||
|
|
LightningCLI()
|
||
|
|
|
||
|
|
from pytorch_lightning.utilities.cli import SaveConfigCallback
|
||
|
|
|
||
|
|
with pytest.raises(NotImplementedError, match=r"SaveConfigCallback.*no longer supported as of v1.9"):
|
||
|
|
SaveConfigCallback()
|
||
|
|
|
||
|
|
from pytorch_lightning.utilities.cli import LightningArgumentParser
|
||
|
|
|
||
|
|
with pytest.raises(NotImplementedError, match=r"LightningArgumentParser.*no longer supported as of v1.9"):
|
||
|
|
LightningArgumentParser()
|
||
|
|
|
||
|
|
from pytorch_lightning.utilities.cli import instantiate_class
|
||
|
|
|
||
|
|
with pytest.raises(NotImplementedError, match=r"instantiate_class.*no longer supported as of v1.9"):
|
||
|
|
instantiate_class()
|