Using pytorch_lightning from lightning import

Hey there,
I saw in the installation guideline, that I should install lightning instead of pytorch_lightning.
Now I have a script where I use a lightning module and a trainer.
How do I import them from the lightning module?


Yes, the name of the package changed from pytorch_lightning to lightning, but all the common imports in the docs and in your existing code should still work.
You can always choose to either

from pytorch_lightning.x.y import z

or (new)

from lightning.pytorch.x.y import z

For example:

from lightning.pytorch import Trainer, LightningModule

# or
import lightning.pytorch as pl


Hope this helps, cheers