def highlight_number(x):
    """Themes numbers in a summary using rich markup.

    We use a separate color for `None`s, e.g. in a layer shape.
    """
    if x is None:
        return f"[color(45)]{x}[/]"
    else:
        return f"[color(34)]{x}[/]"
Themes numbers in a summary using rich markup.

    We use a separate color for `None`s, e.g. in a layer shape.
    
keras/src/utils/summary_utils.py