Imports Lightning loggers

[bertinma]

Hi,
I'm using lightning==2.2.5 and the following imports do not work : from pytorch_lightning.loggers.base import LightningLoggerBase, rank_zero_experiment at line 5 in the file utils/lightning.py

I repalced it by from pytorch_lightning.loggers.logger import Logger, rank_zero_experiment and at line 12 the parent class like this class LabMLLightningLogger(Logger):.

Now it works well :)

Could you please perform those changes ?
Thanks