def treescope_color(self) -> str | tuple[str, str]:
    """Returns a color for this variable in Treescope."""
    raise NotImplementedError(
        "treescope_color must be overridden by subclasses."
    )
Returns a color for this variable in Treescope.
penzai/core/variables.py