Python clear示例

示例#1

def test_v1_7_0_deprecated_lightning_module_summarize(tmpdir):
    from pytorch_lightning.core.lightning import warning_cache

    model = BoringModel()
    model.summarize(max_depth=1)
    assert any("The `LightningModule.summarize` method is deprecated in v1.5" in w for w in warning_cache)
    warning_cache.clear()

示例#2

文件： test_remove_1-5.py 项目： tsumli/pytorch-lightning

def test_v1_5_0_datamodule_setter():
    model = BoringModel()
    datamodule = BoringDataModule()
    with no_deprecated_call(match="The `LightningModule.datamodule`"):
        model.datamodule = datamodule
    from pytorch_lightning.core.lightning import warning_cache
    warning_cache.clear()
    _ = model.datamodule
    assert any("The `LightningModule.datamodule`" in w for w in warning_cache)

示例#3

文件： test_remove_1-6.py 项目： spaparaju/pytorch-lightning

def test_v1_6_0_lightning_module_loaded_optimizer_states_dict():
    from pytorch_lightning.core.lightning import warning_cache
    model = BoringModel()
    _ = model.loaded_optimizer_states_dict
    assert any(
        "The `LightningModule.loaded_optimizer_states_dict` property is deprecated in v1.4"
        in w for w in warning_cache)
    warning_cache.clear()

    model.loaded_optimizer_states_dict = {}
    assert any(
        "The `LightningModule.loaded_optimizer_states_dict` property is deprecated in v1.4"
        in w for w in warning_cache)
    warning_cache.clear()