Non-fatal wandb logging errors

lerobot/src/lerobot/scripts/lerobot_train.py

@@ -499,7 +499,10 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None):
                     upload_thread.start()
                 prev_checkpoint_dir = checkpoint_dir
                 if wandb_logger:
-                    wandb_logger.log_policy(checkpoint_dir)
+                    try:
+                        wandb_logger.log_policy(checkpoint_dir)
+                    except Exception as e:
+                        logging.warning(f"WandB policy logging failed (non-fatal): {e}")
 
             accelerator.wait_for_everyone()