def coreml_precision_to_torch()

in ember/create.py [0:0]

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

def coreml_precision_to_torch(precision: ComputePrecision) -> torch.dtype:
    if precision == ComputePrecision.FLOAT32:
        return torch.float32
    elif precision == ComputePrecision.FLOAT16:
        return torch.float16
    else:
        raise ValueError(f"Unsupported precision: {precision}")