Fix WandbAttachFailedError

✅ Solution

WandbAttachFailedError in pytorch-lightning often arises when Wandb is initialized outside of the main process in a distributed training setting, especially with TPUs, which interferes with proper experiment tracking. To fix this, ensure Wandb is only initialized within the main process (rank 0) by using a conditional check like `if self.trainer.global_rank == 0: wandb.init(...)` or utilize pytorch-lightning's built-in WandbLogger to handle this automatically.