def _arrowtype_to_dtype()

in torcharrow/_interop.py [0:0]

• 18 lines of code
• 10 McCabe index (conditional complexity)

def _arrowtype_to_dtype(t: pa.DataType, nullable: bool) -> dt.DType:
    if pa.types.is_boolean(t):
        return dt.Boolean(nullable)
    if pa.types.is_int8(t):
        return dt.Int8(nullable)
    if pa.types.is_int16(t):
        return dt.Int16(nullable)
    if pa.types.is_int32(t):
        return dt.Int32(nullable)
    if pa.types.is_int64(t):
        return dt.Int64(nullable)
    if pa.types.is_float32(t):
        return dt.Float32(nullable)
    if pa.types.is_float64(t):
        return dt.Float64(nullable)
    if pa.types.is_string(t) or pa.types.is_large_string(t):
        return dt.String(nullable)
    raise NotImplementedError(f"Unsupported Arrow type: {str(t)}")