def get_rank()

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

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

def get_rank() -> int:
    if torch.distributed.is_available() and torch.distributed.is_initialized():
        return torch.distributed.get_rank()
    return 0