diff --git a/docs/source/common/lightning_cli.rst b/docs/source/common/lightning_cli.rst index 7267606b05..a75c77058e 100644 --- a/docs/source/common/lightning_cli.rst +++ b/docs/source/common/lightning_cli.rst @@ -891,6 +891,7 @@ If you need to customize the learning rate scheduler configuration, you can do s .. testcode:: class MyLightningCLI(LightningCLI): + @staticmethod def configure_optimizers(lightning_module, optimizer, lr_scheduler=None): return ...