def _find_device()

in src/kernels/layer.py [0:0]

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

def _find_device(model: "nn.Module") -> Device:
    try:
        param = next(model.parameters())
    except StopIteration:
        raise ValueError(
            "Cannot determine model device, provide as `device` argument to `kernelize`."
        )

    return Device(type=param.device.type)