The idea here is to make the `world_size` consistent between `store` and `dist.init_process_group()`. Previously, it is possible that the `world_size` passed into `dist_init()` is at least `1` but is inconsistent with `self.world_size`, leading to a discrepancy between `store` and `dist.init_process_group()`.