def get_world_size()

in torchrecipes/utils/distributed_utils.py [0:0]

• 4 lines of code
• 3 McCabe index (conditional complexity)

def get_world_size() -> int:
    if torch.distributed.is_available() and torch.distributed.is_initialized():
        return torch.distributed.get_world_size()
    return 1