I think it was deserved. I understand the game situation part of it but it's an awfully big coincidence that a guy who hits a massive home run and celebrates, gets plunked on the first pitch, with a fastball, by a guy who had hit 3 guys in 72 innings this year in his very next PA. Oh, and the Astros were ticked earlier this year at a Garcia celebration following a grand slam.
Could it have been an accident? Sure. But the optics sure don't look good at all. And again, you can certainly sell me on the game situation part (still a 4-2 game, nobody out but a runner on first). And it is the playoffs rather than the regular season. But I've got no issue with the umps taking preventative measures there.