def get_dtype()

in ultravox/utils/device_helpers.py [0:0]

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

def get_dtype(data_type: Optional[str] = None) -> torch.dtype:
    if data_type is None:
        return default_dtype()
    else:
        return (
            torch.bfloat16
            if data_type == "bfloat16"
            else torch.float16 if data_type == "float16" else torch.float32
        )