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.