def METHOD_NAME(self, model: Booster) -> Booster:
    """Run before training is finished."""
    # Update W&B config
    config = model.save_config()
    wandb.config.update(json.loads(config))
    return model