I don't have the exact dimensions of WS-10 especially its newer variants, but all the words out there are quite consistent to indicate that WS-10 is slightly larger than the Al-31, and it has caused problems when they try to install WS-10 on flankers for testing. They eventually solved the problem but it was clear that the fit was tighter and required significant effort to change design
A "semi-official" Chinese source that mentioned this:
There were also articles saying that one of the reasons newer J-10 didn't use WS-10 for mass production was that it made the rear section thicker and increased drag a bit too much (J-10 have more restrictive engine compartment than flankers and are more sensitive to any size change). I cannot find official source confirming this one so you may take it with a grain of salt.
So while they solved the problem for flanker platforms, it may still cause problem for a plane like J-20 because of the reasons I mentioned earlier. It may not be as easy as you'd have imagined. But of course without further information this is at best an educated guess, there's nothing I can say for certain. Maybe the engine compartment of the J-20 is just as spacious or flexible as the flankers, or maybe newer WS-10s are just as compact as the Al-31, who knows?