how to import the ModelCheckpoint callback after using the pip install lightning ?

[shamanez]

Bug description

I install pytorch_lightinng with the pip install lightning==1.9.4.

from pytorch_lightning.callbacks import ModelCheckpoint

The above command works well.

But how can I change it when using the lighting?

from lightning.callbacks import ModelCheckpoint

Gives the following error

ModuleNotFoundError: No module named 'lightning.callbacks'

How to reproduce the bug

from lightning.callbacks import ModelCheckpoint

Error messages and logs

# Error messages and logs here please

ModuleNotFoundError: No module named 'lightning.callbacks'

Environment

Current environment

#- Lightning Version (e.g., 1.94):
#- GPU models and configuration:
#- How you installed Lightning(`pip`):
#- Running environment of LightningApp (e.g. local, cloud):

More info

No response

[awaelchli]

Hi

It needs to be

from lightning.pytorch.callbacks import ModelCheckpoint

You can remember it this way: pytorch_lightning = lightning.pytorch
Hope this helps :)

[shamanez]

Thanks you save a lot of time. Will there be updates in the tutorials as well ?

…

On Mon, 13 Mar 2023 at 4:07 PM, Adrian Wälchli ***@***.***> wrote: Hi It needs to be from lightning.pytorch.callbacks import ModelCheckpoint You can remember it this way: pytorch_lightning = lightning.pytorch Hope this helps :) — Reply to this email directly, view it on GitHub <#17048 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AEA4FGSKUZR76FJAAAO3Z2DW32MP3ANCNFSM6AAAAAAVYPJIEA> . You are receiving this because you authored the thread.Message ID: ***@***.***>

[awaelchli]

Yes absolutely, the work on this has already started (Lightning-AI/tutorials#222).
Let me know if you have any other questions about this.