from ray.air.integrations.wandb import WandbLoggerCallback

WandbLoggerCallback.__module__ = "ray.tune.logger.wandb"