Add a type for test fixture world_size (#61363)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/61363
Test Plan: Sandcastle
Reviewed By: ngimel
Differential Revision: D29561360
fbshipit-source-id: 821217e33adc483b1810585a2b91a2ee416513bd
