def METHOD_NAME(
    partition_interval: int,
    partition_interval_unit: TextClause,
    type_,
):
    """Dispatch to date or datetime function based on the type
    Args:
        partition_field (_type_): _description_
        partition_interval (_type_): _description_
        partition_interval_unit (_type_): _description_
    """
    if isinstance(type_, (sqlalchemy.DATE)):
        return DateAddFn(partition_interval, partition_interval_unit)
    if isinstance(type_, sqlalchemy.DATETIME):
        return DatetimeAddFn(partition_interval, partition_interval_unit)
    return TimestampAddFn(partition_interval, partition_interval_unit)