def get_device()

in code/inference.py [0:0]

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

def get_device():
    device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
    return device