def get_model_family_color(model_family):
  """Returns the canonical color for a model family."""
  # Derived from sns.color_palette("colorblind").
  canonical_colors = {
      "vit": "#0173B2",
      "bit": "#DE8F05",
      "simclr": "#029E73",
      "efficientnet-noisy-student": "#555555",
      "wsl": "#CC78BC",
      "clip": "#CA9161",
      "vgg": "#949494",
      "alexnet": "#949494",
      "mixer": "#D55E00",
      "guo": "#000000",
  }
  assert model_family in canonical_colors, f"Specify color for {model_family}."
  return canonical_colors[model_family]