def get_device()

in src/huggingface_inference_toolkit/utils.py [0:0]

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

def get_device():
    """
    The get device function will return the device for the DL Framework.
    """
    gpu = _is_gpu_available()

    if gpu:
        return 0
    else:
        return -1