def run_upgrade(args): """Run upgrade on master node """ upgrader = Upgrade( args.src, args.dst, PuppetUpgrader(args.src), disable_rollback=args.disable_rollback) upgrader.run()