public function upgrade( $args, $assoc_args ) {

    // Disable post-upgrade redirect.
    remove_action( 'genesis_upgrade', 'genesis_upgrade_redirect' );

    // Call the upgrade function.
    genesis_upgrade();

    WP_CLI::success( __( 'Genesis database upgraded.', 'genesis' ) );

  }