He has been told to follow a strict fitness and shed weight to ensure his return to the Mumbai team.
Prithvi Shaw expressed his frustration after being dropped from Mumbai's squad for the Vijay Hazare Trophy beginning on December 21.
After picking two wickets earlier, Bumrah capped an eventful day by facing 11 balls as the nightwatchman, ducking cleverly and blocking confidently to end with his wicket intact at stumps on Day 2.