def is_time_type(column_type: str) -> bool:
    """
    Check if the column type is a time-related type.
    Determines if the given SQL column type represents a date or timestamp type.
    Args:
        column_type (str): The SQL column type to check.
    Returns:
        bool: True if the column type is time-related, False otherwise.
    """
    return column_type in (
        "DATE",
        "TIMESTAMP",
        "TIMESTAMP_NTZ",
    )

    Check if the column type is a time-related type.
    Determines if the given SQL column type represents a date or timestamp type.
    Args:
        column_type (str): The SQL column type to check.
    Returns:
        bool: True if the column type is time-related, False otherwise.
    
ai/core/src/unitycatalog/ai/core/utils/type_utils.py