Hide Proceed to Checkout in WordPress/WooCommerce

Solution:

Since you only have two roles you’re checking, why not just replace the following line:

if( in_array( 'wholesale', $current_screen_user->roles ) ) {

with

if( in_array( 'wholesale', $current_screen_user->roles ) || in_array( 'us_wholesale', $current_screen_user->roles ) ){

That way your current (presumably working?) functions will run if the current user has either wholesale or us_wholesale as a user role.